EKF-Diagnostic/sensor-anomalies
1
0
Fork 0
generated from dopt-python/py311
Code Issues Pull Requests Actions Packages Projects 1 Releases 6 Wiki Activity

add anomaly threshold as constant

Browse Source
This commit is contained in:
Florian Förster 2025-11-20 15:24:42 +01:00
parent cc5d164ec2
commit f80fbbc1f9
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/dopt_sensor_anomalies/constants.py
View File

@ -9,6 +9,7 @@ THRESHOLD_BW: Final[int] = 63
BACKBONE: Final[str] = "'wide_resnet50_2" BACKBONE: Final[str] = "'wide_resnet50_2"
LAYERS: Final[tuple[str, str]] = ("layer1", "layer2", "layer3") LAYERS: Final[tuple[str, str]] = ("layer1", "layer2", "layer3")
RATIO: Final[float] = 0.01 RATIO: Final[float] = 0.01
ANOMALY_THRESHOLD: Final[float] = 0.2
NUM_VALID_ELECTRODES: Final[int] = 6 NUM_VALID_ELECTRODES: Final[int] = 6
HEATMAP_FILENAME_SUFFIX: Final[str] = "_Heatmap" HEATMAP_FILENAME_SUFFIX: Final[str] = "_Heatmap"

2
src/dopt_sensor_anomalies/detection.py
View File

@ -176,7 +176,7 @@ def infer_image(
output = model(input_tensor) output = model(input_tensor)
anomaly_score = output.pred_score.item() anomaly_score = output.pred_score.item()
anomaly_label = bool(1 if anomaly_score >= 0.2 else 0) anomaly_label = bool(1 if anomaly_score >= const.ANOMALY_THRESHOLD else 0)
anomaly_map = output.anomaly_map.squeeze().cpu().numpy() anomaly_map = output.anomaly_map.squeeze().cpu().numpy()
img_np = np.array(pil_image) img_np = np.array(pil_image)