awacke1's picture
Upload 11 files
0e371d1
raw
history blame
402 Bytes
yolo_config = {
# Basic
'img_size': (416, 416, 3),
'anchors': [12, 16, 19, 36, 40, 28, 36, 75, 76, 55, 72, 146, 142, 110, 192, 243, 459, 401],
'strides': [8, 16, 32],
'xyscale': [1.2, 1.1, 1.05],
# Training
'iou_loss_thresh': 0.5,
'batch_size': 8,
'num_gpu': 1, # 2,
# Inference
'max_boxes': 100,
'iou_threshold': 0.413,
'score_threshold': 0.3,
}