'''
configuration file for training YOLO LITE model.
'''
# NOTICE : ########## Data Related ###########
IMAGE_SIZE = (512, 512)  # (h, w)
# TODO: Change when dataset is changed.

# N_CLASSES = 1
# CLASS_NAME = ['Bean']
# IMG_PATH = '../data/bean_leaf_noweed/images/'
# ANNOTATION_PATH = '../data/bean_leaf_noweed/annotation/test_coco.json'
# BOX_PATH = '../data/bean_leaf_noweed/boxes/'
# OUTPUT_PATH = '../output/bean_leaf_noweed/'
#
N_CLASSES = 2
CLASS_NAME = ['Bean', 'Weed']
IMG_PATH = '../data/bean_leaf/images/'
ANNOTATION_PATH = '../data/bean_leaf/annotation/coco_annotation.json'
BOX_PATH = '../data/bean_leaf/boxes/'
OUTPUT_PATH = '../output/bean_leaf/'

# NOTICE : ########## Model Related ###########
GRID = 8
N_BOXES = 1

# NOTICE : ########## Train Related ###########
DIFFERENT_IMAGE_SIZE = False
LEARNING_RATE = 0.001
weight_decay = 1e-1
DATASET_DIVIDE_RATIO = [0.8, 0.1, 0.1]
TRAINING_BATCH_SIZE = 16
TRAINING_DATA_SHUFFLE = True
TRAINING_NUM_WORKERS = 0
TRAINING_GPU = 2
'''
cpu number.
cuda:0 : 3
cuda:1 : 0
cuda:2 : 2
cuda:3 : 1
'''
TOTAL_EPOCH = 40
LAMBDA_COORD = 5
LAMBDA_NOOBJ = 0.5

# NOTICE: ########## Post Processing ##########
VALID_OUTPUT_THRESHOLD = 0.5
NMS_IOU_THRESHOLD = 0.5

# NOTICE : ######### Plotting ##########
color_map = ['r', 'g', 'c', 'm', 'y', 'k', ]
