search

yingkunwu / R-YOLOv4

This is a PyTorch-based R-YOLOv4 implementation which combines YOLOv4 model and loss function from R3Det for arbitrary oriented object detection.
114 stars 20 forks source link

训练custom数据集的问题 #30

Closed ciomphzx closed 1 year ago

ciomphzx commented 2 years ago

大佬你好，我训练自己的数据集运行train.py文件时出现以下报错： RuntimeError: [enforce fail at inline_container.cc:145] . PytorchStreamReader failed reading zip archive: failed finding central directory pytorch的版本是1.7.0，请问是版本的原因吗

yingkunwu commented 2 years ago

應該不是版本問題，你確定你的路徑是正確的嗎？

ciomphzx commented 2 years ago

我把权重文件换成ryolo.pth可以正常训练了，请问代码从零训练应该从哪里进行修改

yingkunwu commented 2 years ago

你真要全部從頭開始訓練的話 你就在這個function中連結，把54 55行留著其他全部刪掉就可以了。 不過就我的經驗來說，重零開始訓練效果會差非常的多，而且不易收斂。

ciomphzx commented 2 years ago

好的，非常感谢

ciomphzx commented 2 years ago

HRSC2016数据集训练了100epoch，检测时出现了一个目标有重复的框的情况，将检测时的nms_thres设置为0也出现重复框，请问这是什么原因 训练集图片626张 测试集444张

yingkunwu commented 2 years ago

應該是那個目標同時被認為是多重類別？ 例如車子被模型判斷成車子跟卡車

ciomphzx commented 2 years ago

把类别设置为了4种 同一个目标被识别成两种类型

yingkunwu commented 2 years ago

把类别设置为了4种 同一个目标被识别成两种类型

那就沒辦法了。這個模型的在accuracy方面還需要加強，尤其是對比較小的物品。

ciomphzx commented 2 years ago

小目标没有出现这种情况 图像中的小目标被识别出来了 大的目标出现了这种情况