This commit is contained in:
Florian Förster 2025-11-20 08:11:57 +01:00
parent 54babec627
commit 322506bfd0
4 changed files with 719 additions and 690 deletions

File diff suppressed because one or more lines are too long

View File

@ -176,6 +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()
print(f">>> Anomaly score: {anomaly_score}")
anomaly_label = bool(1 if anomaly_score >= 0.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()

Binary file not shown.

Binary file not shown.