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.36k stars 2.84k forks source link

模型需要转为onnx格式固定输入固定输出,但是PCB缺陷检测1个图可能不至1个缺陷,转了几次都不对 #8934

Closed makaiyang closed 2 months ago

makaiyang commented 2 months ago

问题确认 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

是否愿意提交PR? Are you willing to submit a PR?