generated from dopt-python/py311
tests
This commit is contained in:
parent
54babec627
commit
322506bfd0
File diff suppressed because one or more lines are too long
@ -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()
|
||||||
|
|
||||||
|
|||||||
BIN
tests/_models/new/model_left_hand_side.pth
Normal file
BIN
tests/_models/new/model_left_hand_side.pth
Normal file
Binary file not shown.
BIN
tests/_models/new/model_right_hand_side.pth
Normal file
BIN
tests/_models/new/model_right_hand_side.pth
Normal file
Binary file not shown.
Loading…
x
Reference in New Issue
Block a user