报错信息:
RuntimeError: invalid argument 2: size '[224 x 224 x 3]' is invalid for input with 200704 elements at /pytorch/aten/src/TH/THStorage.cpp:84
由于
224*224*3=150528
224*224*4=200704
所以发现问题所在就是 读取图片后,发现不是rgb,是rgba。
意思是 图片带有透明度,所以读取图片时应该
open('xxx.png').convert('RGB')
即可
本文解决了一个常见的PyTorch图片处理错误,当尝试加载一张RGBA格式的图片时,由于PyTorch期望的是RGB格式,导致元素数量不匹配的运行时错误。文章详细解释了错误原因,并提供了解决方案,即在读取图片时使用PIL库的convert('RGB')方法将图片转换为RGB格式。
5522

被折叠的 条评论
为什么被折叠?



