czsnooker的博客

原因输入的图片张量维数不对，报错时输入张量的shape为（3，640，480），pytorch会认为有三张图片，每张图片张量的shape为（640，480），但是它要求图像格式为[C, H, W]（3，640，480），故报错实际上我们想要表达的是一张图片，图片shape为（3，640，480）所以把shape变成（1，3，640，480）就好了x = x.reshape([1, x.shape[0], x.shape[1], x.shape[2]])加这样一句话就行，x为输入的张量。...

pytorch默认是不适用gpu的，即使我们已经安装好了支持的cuda版本和cudnn，所以要使用gpu需要在程序里面设置一下。步骤import torchuse_gpu = torch.cuda.is_available()定义一个检查GPU是否可用的全局变量加载模型、并设置模式： #下载预训练模型 model = torchvision.models.segmentation.deeplabv3_mobilenet_v3_large( pretrained=T