search

jiangjiajun / PaddleUtils

Some tools to operate PaddlePaddle model
Apache License 2.0
69 stars 25 forks source link

PPYOLOE-L转ONNX输出batch是-1 #1

Open 392875 opened 1 year ago

392875 commented 1 year ago

用了paddle_infer_shape.py进行优化后输出为unk__6(很奇怪，这里不能导出为1嘛？) image

jiangjiajun commented 1 year ago

因为multicalss_nms3这个输出的第一维不是batch的意思，表示是最终有多少个框，而这个框的数量不可知的，具体由框里面的数值做完处理后才能知道

392875 commented 1 year ago

嗯嗯好的