admin/momentry_core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: trace debug — show Stranger_NNN for unnamed traces instead of unknown

Browse Source
This commit is contained in:
Accusys
2026-05-14 15:12:21 +08:00
parent 8f013cbdbc
commit c90394897d
42 changed files with 1806 additions and 1722 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/yolo_processor.py
View File

@@ -270,19 +270,19 @@ def process_yolo(
# Load YOLO model (prefer CoreML for ANE acceleration, fallback to PyTorch)
model_path_mlpackage = os.path.join(
os.path.dirname(os.path.abspath(__file__)), "..", "yolov5nu.mlpackage"
os.path.dirname(os.path.abspath(__file__)), "..", "yolov8s.mlpackage"
)
model_path_pt = os.path.join(
os.path.dirname(os.path.abspath(__file__)), "..", "yolov5nu.pt"
os.path.dirname(os.path.abspath(__file__)), "..", "yolov8s.pt"
)
if os.path.exists(model_path_mlpackage):
model = YOLO(model_path_mlpackage)
print("YOLO: CoreML model loaded (4.5x ANE acceleration)")
print("YOLO: CoreML model loaded (YOLOv8s, ANE accelerated)")
elif os.path.exists(model_path_pt):
model = YOLO(model_path_pt)
print("YOLO: PyTorch model loaded")
print("YOLO: PyTorch model loaded (YOLOv8s)")
else:
model = YOLO("yolov5nu.pt") # will auto-download
model = YOLO("yolov8s.pt") # will auto-download
# Get video info
import cv2