-
Notifications
You must be signed in to change notification settings - Fork 28
Open
Description
我用原作者的pytorch2onnx导出onnx文件后,进行加载识别时出现以下错误
Traceback (most recent call last):
File "E:\MyWork\Yolo-FastestV2\dnn.py", line 112, in <module>
outputs = model.detect(srcimg)
File "E:\MyWork\Yolo-FastestV2\dnn.py", line 91, in detect
outputs[top:top + length, 0:2] = (outs[row_ind:row_ind + length,
ValueError: operands could not be broadcast together with shapes (1,2,22,95) (484,2)
当我调试时,发现detect函数中的len(outs)为1,我觉得这是不合理的,您是否可以上传以下pytorch转onnx的代码,或者看一下如何修改bug,我将万分感激!!!
Metadata
Metadata
Assignees
Labels
No labels