from easydict import EasyDict as edict


# init
__C_VISDRONE = edict()

cfg_data = __C_VISDRONE

#__C_VISDRONE.TRAIN_SIZE = (1080,1920)
__C_VISDRONE.TRAIN_SIZE = (512,1024)
__C_VISDRONE.DATA_PATH = './datasets/ProcessedData/VisDrone/'
__C_VISDRONE.TRAIN_LST = 'train.txt'
__C_VISDRONE.VAL_LST =  'val.txt'
__C_VISDRONE.VAL4EVAL = 'val_gt_loc.txt'

__C_VISDRONE.MEAN_STD = ([0.43652326138271863, 0.44653884841207264, 0.4339871961927452], [0.24028930696174566, 0.23616189927050205, 0.24054449751104])

__C_VISDRONE.LABEL_FACTOR = 1
__C_VISDRONE.LOG_PARA = 1.
__C_VISDRONE.RESUME_MODEL = ''#model path
__C_VISDRONE.TRAIN_BATCH_SIZE = 1 #imgs

__C_VISDRONE.VAL_BATCH_SIZE = 1 # must be 1
