pnicewiczoig commited on
Commit
ccfdf7d
·
1 Parent(s): 3b57fcd

Update settings.py

Browse files
Files changed (1) hide show
  1. settings.py +5 -4
settings.py CHANGED
@@ -19,13 +19,13 @@ IMAGE = 'Image'
19
  VIDEO = 'Video'
20
  WEBCAM = 'Webcam'
21
  RTSP = 'RTSP'
22
- YOUTUBE = 'YouTube'
23
 
24
- SOURCES_LIST = [IMAGE, VIDEO, WEBCAM, RTSP, YOUTUBE]
25
 
26
  # Images config
27
  IMAGES_DIR = ROOT / 'images'
28
- DEFAULT_IMAGE = IMAGES_DIR / 'office_4.jpg'
29
  DEFAULT_DETECT_IMAGE = IMAGES_DIR / 'office_4_detected.jpg'
30
 
31
  # Videos config
@@ -47,6 +47,7 @@ VIDEOS_DICT = {
47
  MODEL_DIR = ROOT / 'weights'
48
  DETECTION_MODEL = MODEL_DIR / 'yolov8n.pt'
49
  SEGMENTATION_MODEL = MODEL_DIR / 'yolov8n-seg.pt'
 
50
 
51
  # Webcam
52
- WEBCAM_PATH = 0
 
19
  VIDEO = 'Video'
20
  WEBCAM = 'Webcam'
21
  RTSP = 'RTSP'
22
+ HTTP = 'HTTP/HTTPS'
23
 
24
+ SOURCES_LIST = [IMAGE, VIDEO, WEBCAM, RTSP, HTTP]
25
 
26
  # Images config
27
  IMAGES_DIR = ROOT / 'images'
28
+ DEFAULT_IMAGE = IMAGES_DIR / 'pdc.jpg'
29
  DEFAULT_DETECT_IMAGE = IMAGES_DIR / 'office_4_detected.jpg'
30
 
31
  # Videos config
 
47
  MODEL_DIR = ROOT / 'weights'
48
  DETECTION_MODEL = MODEL_DIR / 'yolov8n.pt'
49
  SEGMENTATION_MODEL = MODEL_DIR / 'yolov8n-seg.pt'
50
+ POSE_MODEL = MODEL_DIR / 'yolov8n-pose.pt'
51
 
52
  # Webcam
53
+ WEBCAM_PATH = 0