_base_ = ["../_base_/base_static.py", "../../_base_/backends/onnxruntime.py"]

onnx_config = dict(
    input_shape=[320, 320],
)

codebase_config = dict(
    # For later TensorRT inference, the number of output boxes needs to be as stable as possible, 
    # because a drop in the box count leads to a re-optimization which takes a lot of time,
    # therefore sort out low confidence boxes outside the model and reduce the maximum number
    # of output boxes to the smallest usable value.
    post_processing=dict(
        score_threshold=0.0,
        confidence_threshold=0.0,
        iou_threshold=0.3,
        max_output_boxes_per_class=10,
    ),
)