settings.py

# MAIN SETTINGS
# Max age of tracks before they disappear
MAX_AGE = 10
# Show the marked video in real time
SHOW = True
# Minimum amount of associations necessary to mark a track
SHOW_MIN_UPDATES = 5
# Use different colors for drawing tracks which have been associated and not
SHOW_UPDATES = False
# Draw predicted positions for tracks
SHOW_KALMAN_FILTER = False
OUTPUT_PATH = 'results/result.mp4'
SAVE_TO_JSON = True
JSON_OUTPUT_PATH = 'results/result.json'

# COST MATRIX SETTINGS
MAHALANOBIS_THRESHOLD = 9.4877
SMALLEST_COSINE_THRESHOLD = 0.5
# Kalman filter settings
TIME_STEP = 1
KALMAN_NOISE_COVARIANCE_Q = [1, 1, 0.01, 0.01, 10, 10, 0.001, 0.001]
KALMAN_NOISE_COVARIANCE_R = [1, 1, 0.01, 0.01]
# ResNet settings
MAX_DESCRIPTORS = 100
RESNET_RESIZE_WIDTH = 224
RESNET_RESIZE_HEIGHT = 224
RESNET_RESIZE_SIZE = (RESNET_RESIZE_WIDTH, RESNET_RESIZE_HEIGHT)

# DETECTION SETTINGS
YOLO_WEIGHTS_PATH = 'yolo/yolov3.weights'
YOLO_CONFIG_PATH = 'yolo/yolov3.cfg'
YOLO_CLASSES_IDENTIFIERS = [0]
CONFIDENCE_THRESHOLD = 0.5
NMS_THRESHOLD = 0.4
YOLO_RESIZE_WIDTH = 416
YOLO_RESIZE_HEIGHT = 416
YOLO_RESIZE_SIZE = (YOLO_RESIZE_WIDTH, YOLO_RESIZE_HEIGHT)