yolov3訓練voc數據集報錯RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1.

1.代碼來自:https://github.com/eriklindernoren/PyTorch-YOLOv3; 2.主要原因是yolov3輸出三個分支,voc類別只有20類,因爲只改動了一個分支,所以出現最後三個分支的輸出張量因爲維度不對,而導致無法拼接; 3.解決方法:在yolov3.cfg中更改配置,共有三處: 第一處: 第二處: 第三處: 以上三處改動地方都一樣,convolutiona
相關文章
相關標籤/搜索