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

fix type issue for anomaly label

Browse Source
This commit is contained in:
Florian Förster 2025-11-10 14:21:45 +01:00
parent 9e6daa01ea
commit 08f70ee672
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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 = 1 if anomaly_score >= .2 else 0 anomaly_label = bool(1 if anomaly_score >= 0.2 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)