Upload agents/sensor_agent.py with huggingface_hub
Browse files- 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)
|