YuZhouShui

泡泡老师，是这样的，我们在使用自制数据集训练文件，读取预训练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,...