P274书上的代码,这里是从ipynb文件中按顺序复制来的:
使用到的图片如下:
代码:
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import PIL
import torch
from torchvision import transforms
import torchvision
model = torchvision.models.segmentation.fcn_resnet101(pretrained=True)
model.eval()
## 读取照片
image = PIL.Image.open("Zdata/img1.jpg")
# 显示图像
plt.imshow(image)
plt.show()
## 图像预处理,转为0-1之间,标准化处理
image_transf = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize(mean = [0.485, 0.456,0.406],
std = [0.229, 0.224, 0.225])
])
image_tensor = image_transf(image).unsqueeze(0)
# 将图像丢入模型进行处理
output