pytorch的torchvision.models模块fasterrcnn目标检测之一

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=[0.229