模型需要转为onnx格式固定输入固定输出,但是PCB缺陷检测1个图可能不至1个缺陷,转了几次都不对
makaiyang opened this issue · 0 comments
问题确认 Search before asking
需求描述 Feature Description
模型需要转为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 data type: 1
INFO:root:input tensor with shape:[1, 3, 640, 640]
INFO:root:input tensor name: scale_factor
INFO:root:input tensor data type: 1
INFO:root:input tensor with shape:[1, 2]
INFO:root:output tensor name: multiclass_nms3_0.tmp_0
INFO:root:output tensor data type: 1
INFO:root:output tensor with shape:[-1, 6]
INFO:root:output tensor name: multiclass_nms3_0.tmp_2
INFO:root:output tensor data type: 6
INFO:root:output tensor with shape:[-1]
Process finished with exit code 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
是否愿意提交PR Are you willing to submit a PR?
- Yes I'd like to help by submitting a PR!