Spaces:
Running
Running
GitHub Actions
commited on
Commit
·
e2c48cc
1
Parent(s):
1c19e5c
Auto-deploy from GitHub Actions: b49256fe169f62429a69023defd34941308b4335
Browse files- .last_deploy_timestamp +1 -1
- config.py +8 -2
.last_deploy_timestamp
CHANGED
|
@@ -1 +1 @@
|
|
| 1 |
-
2025-11-
|
|
|
|
| 1 |
+
2025-11-20 06:33:37
|
config.py
CHANGED
|
@@ -1,11 +1,17 @@
|
|
| 1 |
import json
|
| 2 |
from pathlib import Path
|
| 3 |
from dataclasses import dataclass
|
|
|
|
| 4 |
|
| 5 |
import numpy as np
|
| 6 |
from ultralytics import YOLO
|
| 7 |
|
| 8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
class Config:
|
| 10 |
MODELS_DIR: Path = Path('models')
|
| 11 |
MODELS_DIR.mkdir(exist_ok=True)
|
|
@@ -29,7 +35,8 @@ class Config:
|
|
| 29 |
TRACKER_NAMES: list[str] = list(TRACKERS.keys())
|
| 30 |
WEBCAM_TIME_LIMIT: int = 60
|
| 31 |
WEBCAM_MAX_WINDOW_WIDTH: int = 600
|
| 32 |
-
|
|
|
|
| 33 |
|
| 34 |
@dataclass
|
| 35 |
class DetectConfig:
|
|
@@ -39,7 +46,6 @@ class DetectConfig:
|
|
| 39 |
iou: float
|
| 40 |
detect_mode: str
|
| 41 |
tracker_name: str
|
| 42 |
-
|
| 43 |
verbose: bool = False
|
| 44 |
save_image_predicts: bool = False
|
| 45 |
save_video_predicts: bool = True
|
|
|
|
| 1 |
import json
|
| 2 |
from pathlib import Path
|
| 3 |
from dataclasses import dataclass
|
| 4 |
+
from enum import Enum
|
| 5 |
|
| 6 |
import numpy as np
|
| 7 |
from ultralytics import YOLO
|
| 8 |
|
| 9 |
|
| 10 |
+
class WebcamMode(Enum):
|
| 11 |
+
GRADIO = 'gradio'
|
| 12 |
+
WEBRTC = 'webrtc'
|
| 13 |
+
|
| 14 |
+
|
| 15 |
class Config:
|
| 16 |
MODELS_DIR: Path = Path('models')
|
| 17 |
MODELS_DIR.mkdir(exist_ok=True)
|
|
|
|
| 35 |
TRACKER_NAMES: list[str] = list(TRACKERS.keys())
|
| 36 |
WEBCAM_TIME_LIMIT: int = 60
|
| 37 |
WEBCAM_MAX_WINDOW_WIDTH: int = 600
|
| 38 |
+
WEBCAM_MODE: WebcamMode = WebcamMode.WEBRTC # WebcamMode.WEBRTC or WebcamMode.GRADIO
|
| 39 |
+
|
| 40 |
|
| 41 |
@dataclass
|
| 42 |
class DetectConfig:
|
|
|
|
| 46 |
iou: float
|
| 47 |
detect_mode: str
|
| 48 |
tracker_name: str
|
|
|
|
| 49 |
verbose: bool = False
|
| 50 |
save_image_predicts: bool = False
|
| 51 |
save_video_predicts: bool = True
|