Skip to content

代码bug #1

@violet2422

Description

@violet2422

我用原作者的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

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions