Spaces:
Paused
Paused
fixed not finding weights
Browse files- 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"))
|