-
Notifications
You must be signed in to change notification settings - Fork 123
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
请问使用的数据集图片尺寸是多少呢,predict失败,无文本框 #32
Comments
即便使用提供的ctpn_keras_weights.pth.tar,文本框依旧是[ ],请问是我漏了某个地方需要修改吗 |
@xieguoxuan 我想跟你resize的方式有关系,有没有将test使用的图像也使用同样的resize方式试试 |
有的,resize我是把train和test的一起了,都是640480。如果是resize的问题,那请问你发的那个ctpn_keras_weights可以直接用来做predict吗,那个predict也是没有出现结果。不管是640480还是原始图片大小我都试过。 |
ctpn_keras_weights不能直接predict,只是用来finetune的时候收敛快很多。之前也有感兴趣的好友用过没啥问题呀 |
那请问您有训练好的weight吗,我想测试一下看看是代码没调好还是说我训练部分有问题。可以麻烦您发到 [email protected]吗 |
|
@xieguoxuan 尝试将lr减小,不能增大 |
找不到原数据集就使用了ICDAR2013,由于原尺寸跑这个代码会oom,因此做了resize成640*480。train之后达到这个水准,ctpn_ep18_0.0710_0.1520_0.2228.pth.tar。
接下来使用图片进行predict时,输出没有文本框,代码中print(text)输出为[ ](空?)。是否训练失败,检测不到文本框?也许跟数据集图片尺寸有关
The text was updated successfully, but these errors were encountered: