Hugging Face's logo

Spaces:
chtseng
/
yolo.dog
Sleeping

App Files Community
ch-tseng commited on
Commit
96999c2
·
1 Parent(s): 0f897c1

update

Browse files
Files changed (1) hide show
  1. config.py +3 -2
config.py CHANGED
@@ -30,9 +30,10 @@ SOURCES_LIST = ["Image", "Video", "Webcam"]
30
 
31
 
32
  # DL model config
33
- DETECTION_MODEL_DIR = ROOT / 'models' / 'detection'
 
34
  #YOLOv8n = DETECTION_MODEL_DIR / "yolov8n.pt"
35
- YOLOv8s = './crowded_human_yolov8s.pt'
36
  #YOLOv8m = DETECTION_MODEL_DIR / "yolov8m.pt"
37
  #YOLOv8l = DETECTION_MODEL_DIR / "yolov8l.pt"
38
  #YOLOv8x = DETECTION_MODEL_DIR / "yolov8x.pt"
 
30
 
31
 
32
  # DL model config
33
+ #DETECTION_MODEL_DIR = ROOT / 'models' / 'detection'
34
+ DETECTION_MODEL_DIR = ROOT
35
  #YOLOv8n = DETECTION_MODEL_DIR / "yolov8n.pt"
36
+ YOLOv8s = 'crowded_human_yolov8s.pt'
37
  #YOLOv8m = DETECTION_MODEL_DIR / "yolov8m.pt"
38
  #YOLOv8l = DETECTION_MODEL_DIR / "yolov8l.pt"
39
  #YOLOv8x = DETECTION_MODEL_DIR / "yolov8x.pt"