import torch
import torchvision.models as models
import torchvision
提前下载好权重文件,加载fasterrcnn模型
net = models.detection.fasterrcnn_resnet50_fpn(pretrained=False)
pthfile = r'G:\Anaconda3\envs\pytorch\Lib\site-packages\torchvision\models\fasterrcnn_resnet50_fpn_coco-258fb6c6.pth'
net.load_state_dict(torch.load(pthfile))
#print(net)
我们用PIL读取一张图片,进去看看
from PIL import Image
img = Image.open("maandgirl.jpg")
img
from torchvision import transforms
transform = transforms.Compose([ #[1]
transforms.Resize(256), #[2]
transforms.CenterCrop(224), #[3]
transforms.ToTensor(), #[4]
transforms.Normalize( #[5]
mean=[0.485, 0.456, 0.406], #[6]
std=