...
Create inputs_outputs.txt file and get the information from yolov5s.onnx via netron tool/webpage. Here is the onnx file:
.View file name yolov5s_onnx.zip
...
Select the three operators within the red box as the output. write --input-size-list and --outputs informations to inputs_outputs.txt:
Code Block | ||
---|---|---|
| ||
--outputs 'onnx::Sigmoid_375 onnx::Sigmoid_419 onnx::Sigmoid_462/model.24/Transpose_output_0 /model.24/Transpose_1_output_0 /model.24/Transpose_2_output_0' |
After completing the above steps, there will be the following files under the yolov5s path:
...