fix(inference): resolve multiple YOLO inference and API issues

This commit is contained in:
2026-01-21 14:48:01 +08:00
parent 1b344aeb2e
commit 1c7190bbb0
5 changed files with 146 additions and 80 deletions

10
main.py
View File

@@ -18,6 +18,16 @@ from prometheus_client import start_http_server
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from ultralytics.engine.results import Boxes as UltralyticsBoxes
def _patch_boxes_ndim():
if not hasattr(UltralyticsBoxes, 'ndim'):
@property
def ndim(self):
return self.data.ndim
UltralyticsBoxes.ndim = ndim
_patch_boxes_ndim()
from api.alarm import router as alarm_router
from api.camera import router as camera_router
from api.roi import router as roi_router