07571026
07571026
Hi, I had a similar issue to yours, and I solved it. ``` Traceback (most recent call last): File "d:\onnx2torch\convert.py", line 6, in torch_model = convert(model) File "c:\Python\Python39\pyenv\lib\site-packages\onnx2torch\converter.py", line 110,...
我也遇到一樣的報錯,錯誤原因是list裡面含有兩組以上的array時,將list變成轉成np.array時,裡面的array的維度必須相同,否則就會出錯 因為dt_boxes_new裡面有兩組以上的array,並且他們的shape[0]不同,所以出錯 我調整的方式你參考一下 根據你的錯誤是在 predict_det.py 的 line 261 det_boxes = np.array(dt_boxes_new) 請把這行備註掉,並且把下面的 return det_boxes 備註起來 改成 return det_boxes_new 實際修改如下圖