修改版本:https://github.com/pprp/captcha_identify.torch
pprp opened this issue · 0 comments
pprp commented
@dee1024 感谢您的工作,我研究了以下您的代码,针对您的库中的不足之处进行了部分改进和个性化处理。
这是链接:https://github.com/pprp/captcha_identify.torch
使用您的库:
training set : 10k
test set: 1k
达不到96%的要求,请教您具体是哪个地方有问题?
另外,我主要的改动的点在以下几个方面:
- 添加了更多torchvision中支持的模型
- 改了一下文件的名称
- 支持了GPU,当然cpu也可以
- 添加了以下功能:训练完成一个epoch之后进行测试,(需要保证test和train中的模型一致)
- 添加了以下功能:将每次得到的测试结果写入results.txt文件,运行torch_util.py得到results.png可视化准确率。
- RES152为基础网络进行训练,混合数字和大写字符只能达到94%,还达不到原作者的识别率,希望得到赐教
目前最好的是RES101,RES152为基础的网络,可以达到94%。