错误展示,RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimensio
错误解释
错误代码
解决方法
Image.open()
函数可以读取多种格式的图片,可能导致输入图片与模型期望图片的张量大小不匹配。可使用img = Image.open(img_path).convert('RGB'),将图片转化为RGB格式。
错误展示,RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimensio
错误解释
错误代码
解决方法
Image.open()
函数可以读取多种格式的图片,可能导致输入图片与模型期望图片的张量大小不匹配。可使用img = Image.open(img_path).convert('RGB'),将图片转化为RGB格式。