anky2002 commited on
Commit
57eeb27
·
verified ·
1 Parent(s): 6c17bfb

Upload agents/sensor_agent.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. agents/sensor_agent.py +2 -2
agents/sensor_agent.py CHANGED
@@ -227,7 +227,7 @@ ALL_TESTS=[s01_prnu_uniformity,s02_prnu_correlation,s03_noise_model,s04_bayer,s0
227
  s11_color_matrix,s12_quantization,s13_bit_depth,s14_saturation_clipping,
228
  s15_noise_spatial_freq,s16_green_imbalance,s17_noise_autocorrelation,s18_demosaic_interpolation]
229
 
230
- def run_sensor_agent(img):
231
  from agents.utils import run_agent_tests
232
- findings, avg, conf, fail, rat = run_agent_tests(ALL_TESTS, img, "Sensor Characteristics Agent")
233
  return AgentEvidence("Sensor Characteristics Agent",np.clip(avg,-1,1),conf,fail,rat,findings)
 
227
  s11_color_matrix,s12_quantization,s13_bit_depth,s14_saturation_clipping,
228
  s15_noise_spatial_freq,s16_green_imbalance,s17_noise_autocorrelation,s18_demosaic_interpolation]
229
 
230
+ def run_sensor_agent(img, modality_adjustments=None):
231
  from agents.utils import run_agent_tests
232
+ findings, avg, conf, fail, rat = run_agent_tests(ALL_TESTS, img, "Sensor Characteristics Agent", modality_adjustments)
233
  return AgentEvidence("Sensor Characteristics Agent",np.clip(avg,-1,1),conf,fail,rat,findings)