The reason for the error: the picture format was not converted, and the color space of different picture formats is different
[Solved] RuntimeError:stack expects each tensor to be equal size,but got [3,224,244] at entry 0 and [1,224,244] at entry 7
Leave a reply