19 lines
659 B
Python
19 lines
659 B
Python
_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,
|
|
),
|
|
)
|