_base_ = ['./base_static.py'] | |
codebase_config = dict( | |
type='mmyolo', | |
task='ObjectDetection', | |
model_type='end2end', | |
post_processing=dict( | |
score_threshold=0.05, | |
confidence_threshold=0.005, | |
iou_threshold=0.5, | |
max_output_boxes_per_class=200, | |
pre_top_k=5000, | |
keep_top_k=100, | |
background_label_id=-1), | |
module=['mmyolo.deploy']) | |
backend_config = dict(type='onnxruntime') | |