fix(inference): resolve multiple YOLO inference and API issues
This commit is contained in:
10
main.py
10
main.py
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user