YOLO / yolo /config /task /inference.yaml
henry000's picture
🔧 [Add] MaxBox for NMS output
1e35163
raw
history blame contribute delete
235 Bytes
task: inference
fast_inference: # onnx, trt, deploy or Empty
data:
source: demo/images/inference/image.png
image_size: ${image_size}
data_augment: {}
nms:
min_confidence: 0.5
min_iou: 0.5
max_bbox: 300
save_predict: True