AliZak commited on
Commit
21c159e
·
verified ·
1 Parent(s): 5671fc1

fixed not finding weights

Browse files
Files changed (1) hide show
  1. perception_roi_server.py +1 -1
perception_roi_server.py CHANGED
@@ -34,7 +34,7 @@ from fastapi.responses import FileResponse, StreamingResponse, JSONResponse
34
  from ultralytics import YOLO
35
 
36
  DEFAULT_WEIGHTS = os.environ.get("YOLO_WEIGHTS", "yolov8s.pt")
37
- WEIGHTS_DIR = os.environ.get("WEIGHTS_DIR", "")
38
  DEFAULT_CONF = float(os.environ.get("YOLO_CONF", "0.25"))
39
  DEFAULT_DEVICE = os.environ.get("YOLO_DEVICE", "auto")
40
  FAST_DETECT_SCALE = float(os.environ.get("FAST_DETECT_SCALE", "0.65"))
 
34
  from ultralytics import YOLO
35
 
36
  DEFAULT_WEIGHTS = os.environ.get("YOLO_WEIGHTS", "yolov8s.pt")
37
+ WEIGHTS_DIR = os.environ.get("WEIGHTS_DIR", os.path.dirname(__file__))
38
  DEFAULT_CONF = float(os.environ.get("YOLO_CONF", "0.25"))
39
  DEFAULT_DEVICE = os.environ.get("YOLO_DEVICE", "auto")
40
  FAST_DETECT_SCALE = float(os.environ.get("FAST_DETECT_SCALE", "0.65"))