Pytorch Detectron2 图像预处理

detectron2/data/detection_utils.py

detectron2/data/transforms.py 

 

from detectron2.config import get_cfg
from detectron2.data import detection_utils
from detectron2.data import transforms as T
from PIL import Image
from torchvision import transforms
device = torch.device('cuda')

# 指定需要的变换
transform=transforms.Compose([transforms.Resize(256), transforms.CenterCrop(224), transforms.ToTensor(), 
                transforms.Normalize(mean=[0.485,0.456,0.406], std=[0.229,0.224,0.225])])

# 读取图像并施加变换和转换格式
img = Image.open(root)
img = transform(img).unsqueeze(0)

# 建立模型
cfg = get_cfg()
model = build_resnet(cfg)

# 进行预测
o = model(img)