报错原因:没有转换图片格式,不同的图片格式的颜色空间不同
参考:
https://blog.csdn.net/missyougoon/article/details/85331493
RuntimeError:stack expects each tensor to be equal size,but got [3,224,244] at entry 0 and [1,224,244] at entry 7
原文:https://www.cnblogs.com/zhaolimin/p/15110451.html