PaddlePaddle / PaddleDetection

Object Detection toolkit based on PaddlePaddle. It supports object detection, instance segmentation, multiple object tracking and real-time multi-person keypoint detection.
Apache License 2.0
12.83k stars 2.89k forks source link

ppyolo+s转onnx #9198

Closed qiuzhewei closed 3 weeks ago

qiuzhewei commented 3 weeks ago

问题确认 Search before asking

请提出你的问题 Please ask your question

你好, 我训练了一个ppyolo+s的模型,准备把模型转成onnx格式,在使用export_model.py的时候我显示指定了export.nms=False,但是用生成的模型转onnx的时候还是提示 multiclass_nms3_0.tmp_1] [WARNING] Due to the operator multiclass_nms3, the exported ONNX model will only supports inference with input batch_size == 1. 另外我在runtime.yml里面手动改了也没用,NMS模型还是包含了。但是之前转picodet的时候就没这个问题,是转onnx的时候只能用batchsize=1进行推理吗?

qiuzhewei commented 3 weeks ago

solved