| 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) | |