Verify YOLO heatmap generation in deep acceptance

2026-06-30 13:52:00 +08:00
parent cf920e97c3
commit 43ed767b4f
4 changed files with 58 additions and 9 deletions
--- a/backend/app/modules/results/service.py
+++ b/backend/app/modules/results/service.py
@@ -32,6 +32,9 @@ def result_roots() -> list[Path]:
    upload_root = project / "var" / "uploads" / "datasets"
    if upload_root.exists():
        roots.extend(path for path in upload_root.glob("*/results") if path.is_dir())
+    acceptance_root = project / "var" / "acceptance"
+    if acceptance_root.exists():
+        roots.extend(path for path in acceptance_root.glob("deep_*/yolo_tiny/runs/tiny/HeartMap_Visual") if path.is_dir())
    return roots