报错信息:
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')
即可