Upload config.py
Browse files
config.py
CHANGED
|
@@ -31,7 +31,7 @@ SOURCES_LIST = ["Image", "Video", "Webcam"]
|
|
| 31 |
|
| 32 |
# DL model config
|
| 33 |
DETECTION_MODEL_DIR = ROOT / 'weights' / 'detection'
|
| 34 |
-
YOLOv8n = DETECTION_MODEL_DIR / "yolov8n.pt"
|
| 35 |
YOLOv8s = DETECTION_MODEL_DIR / "yolov8s.pt"
|
| 36 |
YOLOv8m = DETECTION_MODEL_DIR / "yolov8m.pt"
|
| 37 |
YOLOv8l = DETECTION_MODEL_DIR / "yolov8l.pt"
|
|
|
|
| 31 |
|
| 32 |
# DL model config
|
| 33 |
DETECTION_MODEL_DIR = ROOT / 'weights' / 'detection'
|
| 34 |
+
YOLOv8n = DETECTION_MODEL_DIR / "yolov8n-seg.pt"
|
| 35 |
YOLOv8s = DETECTION_MODEL_DIR / "yolov8s.pt"
|
| 36 |
YOLOv8m = DETECTION_MODEL_DIR / "yolov8m.pt"
|
| 37 |
YOLOv8l = DETECTION_MODEL_DIR / "yolov8l.pt"
|