训练自己数据集预测为空

YuZhouShui commented 1 year ago

博主你好，我在使用yolov4+mobilenet和resnet50时使用自己数据集训练，训练过程map有值很小，但是预测图片无法得到结果。在使用VOC数据集时训练一次即可得到较好结果。训练过程之前参考过yolov3可以得到正常跑并预测图片，可以确定该修改的都改了，请问如何改正。

bubbliiiing commented 1 year ago

额……不太理解，使用VOC数据集时训练一次即可得到较好结果，是指voc吗

YuZhouShui commented 1 year ago

泡泡老师，是这样的，我们在使用自制数据集训练文件，读取预训练yolov4ResNET50voc.pth文件，pretrain为false，可以正常训练，但是训练完后打印map只有两个类有值很小，预测测试集图片无法预测，无框。测试了很多次都是这样，一开始以为是数据集问题，就使用voc2007训练，同样读取预训练文件，训练一轮即可正常预测。后面测试多个backbone均是如此。最后将默认优化器SGD更换为Adam，即可正常使用自制数据集训练和完成测试工作。

税宇航 @.***

---原始邮件--- 发件人: @.> 发送时间: 2023年5月7日(周日) 晚上10:53 收件人: @.>; 抄送: @.**@.>; 主题: Re: [bubbliiiing/mobilenet-yolov4-pytorch] 训练自己数据集预测为空 (Issue #56)

额……不太理解，使用VOC数据集时训练一次即可得到较好结果，是指voc吗

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

bubbliiiing commented 1 year ago

哦哦估计是数据集比较小欸

bubbliiiing / mobilenet-yolov4-pytorch

训练自己数据集预测为空 #56