search

ypwhs / captcha_break

验证码识别
MIT License
2.73k stars 684 forks source link

运行winpy/main.py遇到的问题 #61

Open AzureSkyHuHu opened 4 years ago

AzureSkyHuHu commented 4 years ago

我更改了CaptchaDataset 的 getitem 改成读取某个文件夹里的图片 用来测试没问题

但是当用来训练的时候 image 就会报错

因为之前并没接触过这个 所以可能问题描述的不是很清楚 希望有空的时候解答一下

AzureSkyHuHu commented 4 years ago

RuntimeError: stack expects each tensor to be equal size, but got [4] at entry 0 and [5] at entry 7

ypwhs commented 4 years ago

你可以检查一下文件夹里的图片，是否都是RGB图片，尺寸是否一致。

如果不一致，你需要手动写一个 collate 函数，对尺寸不一致的图片进行一个补全操作（padding）。 比如先创建一个 (batch_size, height, max_width, 3) 的矩阵，然后将每张图贴进去。

参考链接：

AzureSkyHuHu commented 4 years ago

已解决 谢谢大佬

Alexuebo commented 3 years ago

大佬您好，为什么我检查了我图片尺寸都是一样的（3x70x200)，还是出现了这种问题呢？以及collate_fn函数具体要怎么操作啊 image

jjw12345 commented 2 years ago

我是训练效果很差。。。