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

src/dopt_sensor_anomalies/detection.py aktualisiert

Browse Source
This commit is contained in:
frasu 2025-11-04 10:24:53 +00:00
parent 7472a128dd
commit 346b905a86
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/dopt_sensor_anomalies/detection.py
View File

@ -141,9 +141,9 @@ def measure_length(
)
x_min = max(min(np.min(c[:, 0, 0]) for c in filtered_cnts) - 20, 0)
x_max = max(np.max(c[:, 0, 0]) for c in filtered_cnts) + 20
x_max = min(max(np.max(c[:, 0, 0]) for c in filtered_cnts) + 20, orig.shape[1])
y_min = max(min(np.min(c[:, 0, 1]) for c in filtered_cnts) - 20, 0)
y_max = max(np.max(c[:, 0, 1]) for c in filtered_cnts) + 20
y_max = min(max(np.max(c[:, 0, 1]) for c in filtered_cnts) + 20, orig.shape[0])
rightmost_x_third = max(filtered_cnts[2][:, 0, 0])
leftmost_x_fourth = min(filtered_cnts[3][:, 0, 0])
@ -176,7 +176,7 @@ def infer_image(
output = model(input_tensor)
anomaly_score = output.pred_score.item()
anomaly_label = output.pred_label.item()
anomaly_label = 1 if anomaly_score >= .2 else 0
anomaly_map = output.anomaly_map.squeeze().cpu().numpy()
img_np = np.array(pil_image)