Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

使用预训练权重训练voc模型,损失减到8,但是测试时检测不到物体,请问对预测图片的处理是否跟训练是一致? #72

Open
diegolitianfu opened this issue Sep 26, 2021 · 5 comments

Comments

@diegolitianfu
Copy link

如题,使用您给的权重可以检测出,但是使用自己训练的模型检测不出来,发现好像预测时对图片的处理跟训练不一致,导致检测不出来?

@bubbliiiing
Copy link
Owner

都是/255哪里不一致啊

@bubbliiiing
Copy link
Owner

你具体的位置标志一下,现在这个库应该是没有不一样的

@diegolitianfu
Copy link
Author

您好,首先非常感谢你的回答。我主要有以下几点疑惑。
1、我指的不是/255,我看dataloader里面,会对训练图像做一些色域扭曲或者翻转图像之类的,但是检测时并不会做这些处理,不知道这对检测结果是否有影响?之前一直有这样的疑问,望解答。
2、我使用您给的权重作为预训练权重,训练集使用voc07+12,一共进行了两次训练,第一次训练训练到第二阶段时我看损失降到8以下,我以为已经很好了,就暂停了训练,但是实际测试结果很差,confidence为0.5时,您给的那张street.jpg一个都检测不出来;第二次训练我让他一直训练,但是发现最终损失没有降下来,最后结束时还是在20. 测试效果也是很差。比较迷惑。
3、未使用预训练权重,从头训练voc模型,使用了mosaic数据增强,训练模型损失可以降到5左右,在voc07test上的map值为58.82%,感觉也不是很好?如果可以的话,想请问一下您训练时都使用了哪些技巧?非常感谢~
祝 好!

@bubbliiiing
Copy link
Owner

1、这个叫数据增强呀……预测增强啥
2、你用了预训练权重效果理论上基本上都是比不用好很多的,我觉得你可能得重下库然后什么都不改来一遍,默认参数就是训练voc的。
3、没用预训练权重的话确实如此,我昨天更新了一下,加了一些注释,你看看吧

@diegolitianfu
Copy link
Author

好的,非常感谢~

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants