基于PP_YOLOE实现PCB缺陷检测模型转onnx格式输出维度可以为1吗

[makaiyang]

模型需要转为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