demo_utils/configuration.py

DISTANCE_THRESHOLD_CENTROID: int = 0.8
DISTANCE_THRESHOLD_BBOX: int = 0.7
MAX_DISTANCE: int = 10000

models_path = {"YOLOv7": "custom_models/yolov7.pt", "YOLOv7 Tiny": "custom_models/yolov7-tiny.pt"}

style = {"Bounding box": "bbox", "Centroid": "centroid"}

examples = {
    "soccer": {"distance_threshold": 0.7, "absolute_path": True, "classes": [0]},
    "oxford_town_center": {"distance_threshold": 0.7, "absolute_path": False, "classes": [0]},
    "traffic_1_A": {"distance_threshold": 0.7, "absolute_path": False, "classes": [2, 3, 5, 7]},
}