_base_ = ["../_base_/base_static.py", "../../_base_/backends/onnxruntime-fp16.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 reduce the maximum number of output boxes to the smallest usable value and sort out
    # low confidence boxes outside the model.
    post_processing=dict(
        score_threshold=0.0,
        confidence_threshold=0.0,
        iou_threshold=0.5,
        max_output_boxes_per_class=10,
    ),
)