你好，yolov5-6转换会报错list index out of range

inisis / brocolli

Everything in Torch Fx

MIT License

341 stars 63 forks source link

Closed ZHEQIUSHUI closed 2 years ago

ZHEQIUSHUI commented 2 years ago

pytorch_graph.py->extract()->line 130

因为不在同一台电脑，以上是出问题的定位，请见谅

ZHEQIUSHUI commented 2 years ago

我用onnx2caffe也会有类似的问题，好像是因为这里，t'ran'stranspose之后，同时作为输入和输出，导致找不到296节点

c5a78f72cb1cbd85ca3425699fcbf66

ZHEQIUSHUI commented 2 years ago

74591ab055905e16f57f9b5d1983d49

inisis commented 2 years ago

pytorch_graph.py->extract()->line 130

因为不在同一台电脑，以上是出问题的定位，请见谅

yolo的话你可以试下这个 https://github.com/zhiqwang/yolov5-rt-stack ，一站式部署，不用自己转模型了。

inisis commented 2 years ago

pytorch_graph.py->extract()->line 130

因为不在同一台电脑，以上是出问题的定位，请见谅

你好，brocolli现在支持yolov5了，参见test/test_nets.py test_yolov5部分。

ZHEQIUSHUI commented 2 years ago

谢谢您