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