使用MSRA-TD500训练,测试时没有结果
shaoming20798 opened this issue · 6 comments
shaoming20798 commented
Hi,作者你好。
首先,很感谢你的分享。我按照你写的训练方法,训练了MSRA-TD500数据,跑rotation_demo.py时,没有输出结果。调试看到,框的置信度都很低,都不超过0.1,所以都被滤掉了。看了下训练日志,开始loss是1.6,迭代20W次后是0.35。看上去应该是收敛了。但是不清楚为什么没有检测出框。
请问,
1.这种情况下是不是真的收敛了,还是哪些地方需要调整下。
2.训练数据需不需要做什么变换,我用的是MSRA-TD500原始标注locations.xml,没有做任何的二次处理。
BaofengZan commented
@shaoming20798 你好,这个问题你最后查出来原因了么?
shaoming20798 commented
@BaofengZan ,是框的分数太低,都给过滤掉了,不知道什么原因
BaofengZan commented
@shaoming20798 能否加个qq,关于这个有好多疑问,可以交流下。
chenmyzju87 commented
你好 我也出现这个问题,训练的模型测试没有结果 无序的框 分值很低
chenmyzju87 commented
@BaofengZan 能否一起交流探讨
BaofengZan commented
@chenmyzju87 我最后发现 我的原因是训练的问题,训练数据过少,你可以对数据增强,或者看论文里面将两个数据集合在一起一起训练,再去测试,效果会好点。