模型需要转为onnx格式固定输入固定输出,但是PCB缺陷检测1个图可能不至1个缺陷,转了几次都不对
makaiyang opened this issue · 0 comments
问题确认 Search before asking
Bug组件 Bug Component
Deploy
Bug描述 Describe the Bug
模型需要转为onnx格式固定输入固定输出,但是PCB缺陷检测1个图可能不至1个缺陷,转了几次都不对
D:\ancanda\miini\envs\onnx\python.exe D:\ocr\Paddle2ONNX-develop\onnx_test.py
INFO:root:onnx model with info:8
INFO:root:onnx model with info:PaddlePaddle
INFO:root:onnx model with info:
INFO:root:input tensor name: image
INFO:root:input tensor 数据类型: 1
INFO:root:输入张量,形状:[1, 3, 640, 640]
INFO:根:输入张量名称:scale_factor
INFO:root:输入张量 数据类型:1
INFO:root:输入张量 shape:[1, 2]
INFO:root:输出张量名称:multiclass_nms3_0.tmp_0
INFO:root:输出张量数据类型:1
INFO:root:输出张量 shape:[-1, 6]
INFO:root:输出张量名称:multiclass_nms3_0.tmp_2
信息:根:输出张量 数据类型:6
INFO:root:输出张量 shape:[-1]
进程完成,退出代码为 0
使用的命令:(onnx) PS D:\ocr\Paddle2ONNX-develop> paddle2onnx --model_dir test --model_filename model.pdmodel --params_filename model.pdiparams --opset_version 11 --save_file PCB_detect.onnx --inp
ut_shape_dict “{'image':[1,3,640,640],'scale_factor':[1,2]}” --enable_onnx_checker True
复现环境 Environment
操作系统:windows
Bug描述确认 Bug description confirmation
- 我确认已经提供了Bug复现步骤、代码改动说明、以及环境信息,确认问题是可以复现的。I confirm that the bug replication steps, code change instructions, and environment information have been provided, and the problem can be reproduced.
是否愿意提交PR? Are you willing to submit a PR?
- 我愿意提交PR!I'd like to help by submitting a PR!