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
笔记-pytorch
猜你喜欢
转载自blog.csdn.net/weixin_41874898/article/details/121931475
今日推荐
周排行