首页
Python
Java
前端
数据库
Linux
Chatgpt专题
开发者工具箱
mat2专题
解决:RuntimeError: mat1 and mat2 shapes cannot be multiplied(单张图片输入情况,可参考)
在练习使用pytorch加载模型,识别图片时,出现了这一问题。 解决方法:使用torch.reshape()将输入数据格式改成与网络相符的格式。 详细过程: 报错代码: import torchimport torchvisionfrom PIL import Imagefrom model import *image = Image.open("./img/dog.png")tra
阅读更多...