YOLO_CONFIG.N_CLASSES = 1 YOLO_CONFIG.NMS_TOPK = 150 YOLO_CONFIG.NMS_THRESHOLD = 0.45 YOLO_CONFIG.SCORE_THRESHOLD = 0.01 YOLO_CONFIG.RESTORE_PATH = '/home/huydao/Source/LogoDetection/data/weights/yolov3.ckpt' YOLO_CONFIG.SAVE_DIR = '/home/huydao/Source/LogoDetection/checkpoints' YOLO_CONFIG.BATCH_SIZE = 8 YOLO_CONFIG.TOTAL_EPOCHS = 10 YOLO_CONFIG.TRAIN_EVALUATION_STEP = 10 YOLO_CONFIG.VAL_EVALUATION_STEP = 2 YOLO_CONFIG.SAVE_EPOCH = 10 YOLO_CONFIG.BATCH_NORM_DECAY = 0.99 WEIGHT_DECAY = 5e-4 YOLO_CONFIG.CLASSES_PATH = '' YOLO_CONFIG.ANCHORS_PATH = '' YOLO_CONFIG.TRAIN_EXAMPLES = len(open(YOLO_CONFIG.TRAIN_PATH,'r').readlines()) YOLO_CONFIG.VAL_EXAMPLES = len(open(YOLO_CONFIG.VAL_PATH,'r').readlines()) YOLO_CONFIG.TRAIN_BATCH_NUM = int(math.ceil(float(YOLO_CONFIG.TRAIN_EXAMPLES)/YOLO_CONFIG.BATCH_SIZE)) YOLO_CONFIG.ANCHORS = parse_anchors(YOLO_CONFIG.ANCHORS_PATH) YOLO_CONFIG.MULTI_SCALE_TRAIN = True YOLO_CONFIG.USE_MIX_UP = True YOLO_CONFIG.LETTER_BOX_RESIZE = True