Open YuZhouShui opened 1 year ago
额……不太理解,使用VOC数据集时训练一次即可得到较好结果,是指voc吗
泡泡老师,是这样的,我们在使用自制数据集训练文件,读取预训练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: @.***>
哦哦 估计是数据集比较小欸
博主你好,我在使用yolov4+mobilenet和resnet50时使用自己数据集训练,训练过程map有值很小,但是预测图片无法得到结果。在使用VOC数据集时训练一次即可得到较好结果。训练过程之前参考过yolov3可以得到正常跑并预测图片,可以确定该修改的都改了,请问如何改正。