def default_transform(size):
transform = transforms.Compose([
transforms.Scale(size),
transforms.CenterCrop(size),
transforms.ToTensor(),
transforms.Normalize(mean=[0.485, 0.456, 0.406], # resnet imagnet
std=[0.229, 0.224, 0.225])
])
return transform
评论列表
文章目录