generated from dopt-python/py311
Compare commits
4 Commits
main
...
verify_res
| Author | SHA1 | Date | |
|---|---|---|---|
| 9cd8457762 | |||
| 322506bfd0 | |||
| 54babec627 | |||
| d7feb53b90 |
@ -5,7 +5,7 @@ description = "anomaly detection for sensor images for quality assurance process
|
|||||||
authors = [
|
authors = [
|
||||||
{name = "d-opt GmbH (resp.: Florian Foerster)", email = "f.foerster@d-opt.com"},
|
{name = "d-opt GmbH (resp.: Florian Foerster)", email = "f.foerster@d-opt.com"},
|
||||||
]
|
]
|
||||||
dependencies = ["imutils>=0.5.4", "dopt-basics>=0.2.4", "numpy>=2.2.6", "anomalib[vlm_clip]>=2.1.0", "open-clip-torch>=3.2.0"]
|
dependencies = ["imutils>=0.5.4", "dopt-basics>=0.2.4", "numpy>=2.2.6", "open-clip-torch>=3.2.0", "anomalib==2.0.0"]
|
||||||
requires-python = "<3.14,>=3.11"
|
requires-python = "<3.14,>=3.11"
|
||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
license = {text = "LicenseRef-Proprietary"}
|
license = {text = "LicenseRef-Proprietary"}
|
||||||
|
|||||||
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/_archive/model_left_hand_side.pth
Normal file
BIN
tests/_models/_archive/model_left_hand_side.pth
Normal file
Binary file not shown.
BIN
tests/_models/_archive/model_right_hand_side.pth
Normal file
BIN
tests/_models/_archive/model_right_hand_side.pth
Normal file
Binary file not shown.
Binary file not shown.
Binary file not shown.
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