File size: 661 Bytes
a090915 e2af51e a090915 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import os
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
DETECT_MODEL = os.path.join(BASE_DIR, "models", "yolov8n.pt")
POSE_MODEL = os.path.join(BASE_DIR, "models", "yolov8n-pose.pt")
MAIN_MODEL = os.path.join(BASE_DIR, "models", "xgb_model.pkl")
# Thresholds and params
CONF_THRESHOLD = float(os.getenv("CONF_THRESHOLD", 0.3))
INACTIVE_TIMEOUT = int(os.getenv("INACTIVE_TIMEOUT", 30))
FRAME_SKIP = int(os.getenv("FRAME_SKIP", 2))
INPUT_SIZE = int(os.getenv("INPUT_SIZE", 640))
# Paths
BASE_DIR = os.path.abspath(os.path.dirname(__file__))
OUTPUT_DIR = os.getenv("OUTPUT_DIR", os.path.join(BASE_DIR, "output"))
os.makedirs(OUTPUT_DIR, exist_ok=True)
|