poulta55 commited on
Commit
954de2e
·
verified ·
1 Parent(s): 2474074

Upload config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -1
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"