Spaces:
Runtime error
Runtime error
File size: 402 Bytes
5c00143 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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,
}
|