mjq11302010044/RRPN

使用MSRA-TD500训练,测试时没有结果

shaoming20798 opened this issue · 6 comments

Hi,作者你好。
首先,很感谢你的分享。我按照你写的训练方法,训练了MSRA-TD500数据,跑rotation_demo.py时,没有输出结果。调试看到,框的置信度都很低,都不超过0.1,所以都被滤掉了。看了下训练日志,开始loss是1.6,迭代20W次后是0.35。看上去应该是收敛了。但是不清楚为什么没有检测出框。
请问,
1.这种情况下是不是真的收敛了,还是哪些地方需要调整下。
2.训练数据需不需要做什么变换,我用的是MSRA-TD500原始标注locations.xml,没有做任何的二次处理。

@shaoming20798 你好,这个问题你最后查出来原因了么?

@BaofengZan ,是框的分数太低,都给过滤掉了,不知道什么原因

@shaoming20798 能否加个qq,关于这个有好多疑问,可以交流下。

你好 我也出现这个问题,训练的模型测试没有结果 无序的框 分值很低

@BaofengZan 能否一起交流探讨

@chenmyzju87 我最后发现 我的原因是训练的问题,训练数据过少,你可以对数据增强,或者看论文里面将两个数据集合在一起一起训练,再去测试,效果会好点。