笔记-pytorch

def preprocessImg(img):
    # preprocess = transforms.Compose([
    #     transforms.ToPILImage(),
    #     transforms.Resize(size=112),  # Pre-trained model uses 140x140 input images
    #     transforms.ToTensor(),
    #     # transforms.Normalize(
    #     #     mean=[0.3880, 0.3880, 0.3880],
    #     #     std=[0.2171, 0.2171, 0.2171]
    #     #     )
    #     ]
    # )
    # img = preprocess(img)
    img = img/255.
    img = (img-0.3880)/0.2171
    img = np.transpose(img,(2, 0, 1))
    img = torch.Tensor(img)
    img = img.to(torch.device("cuda"))

    return img