config.py

import os


WEIGHT_DIR = "weights"
ROOT_DIR = os.path.dirname(os.path.abspath(__file__))