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://blog.csdn.net/missyougoon/article/details/85331493

posted @ 2021-08-06 21:45  赵立敏  阅读(1439)  评论(0编辑  收藏  举报