search

lumina37 / rotate-captcha-crack

CNN预测图片旋转角✨可用于破解旋转验证码
The Unlicense
312 stars 89 forks source link

大佬，代码clone下来后按照您的说明操作报错了呀 #1

Closed 1229984599 closed 1 year ago

1229984599 commented 2 years ago

我丢了一些图片在train里面，数据集结构是这个样子的 image 但是运行train就直接报错了 image

lumina37 commented 2 years ago

图片数量不够？如果你没有改过config.yaml的话那训练集里至少需要有batch_size＝48张图片

lumina37 commented 2 years ago

话说你本科有参加过无线电爱好者协会吗？你这个id和我一个学姐的名字拼音一模一样

1229984599 commented 2 years ago

应该不是吧，数据集我爬了两百张图片的

lumina37 commented 2 years ago

确实是数据集太小的问题，验证用的val_dataloader的batch_size默认为48。如果你的数据集总大小只有200的话就意味着验证集只有10张图，不够填满一个batch。建议换个1000张图的数据集或者为val_dataloader指定一个比较小的batch_size