本文整理匯總了Python中transforms.Normalize方法的典型用法代碼示例。如果您正苦於以下問題：Python transforms.Normalize方法的具體用法？Python transforms.Normalize怎麽用？Python transforms.Normalize使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類transforms的用法示例。
在下文中一共展示了transforms.Normalize方法的3個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: read_image
# 需要導入模塊: import transforms [as 別名]
# 或者: from transforms import Normalize [as 別名]
def read_image(img_path):
"""Keep reading image until succeed.
This can avoid IOError incurred by heavy IO process."""
got_img = False
if not osp.exists(img_path):
raise IOError("{} does not exist".format(img_path))
while not got_img:
try:
img = Image.open(img_path).convert('RGB')
got_img = True
except IOError:
print("IOError incurred when reading '{}'. Will redo. Don't worry. Just chill.".format(img_path))
pass
transform_test = T.Compose([
T.Resize((args.height, args.width)),
T.ToTensor(),
T.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
])
img = transform_test(img)
return img 示例2: get_transform
# 需要導入模塊: import transforms [as 別名]
# 或者: from transforms import Normalize [as 別名]
def get_transform(mode,base_size):
#base_size = 520
#crop_size = 480
crop_size=int(480*base_size/520)
min_size = int((0.5 if mode=='train' else 1.0) * base_size)
max_size = int((2.0 if mode=='train' else 1.0) * base_size)
transforms = []
transforms.append(T.RandomResize(min_size, max_size))
if mode=='train':
transforms.append(T.RandomHorizontalFlip(0.5))
transforms.append(T.RandomCrop(crop_size))
transforms.append(T.ToTensor())
transforms.append(T.Normalize(mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225]))
return T.Compose(transforms) 示例3: __init__
# 需要導入模塊: import transforms [as 別名]
# 或者: from transforms import Normalize [as 別名]
def __init__(self, batch_size, use_gpu, num_workers):
transform = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize((0.1307,), (0.3081,))
])
pin_memory = True if use_gpu else False
trainset = torchvision.datasets.MNIST(root='./data/mnist', train=True, download=True, transform=transform)
trainloader = torch.utils.data.DataLoader(
trainset, batch_size=batch_size, shuffle=True,
num_workers=num_workers, pin_memory=pin_memory,
)
testset = torchvision.datasets.MNIST(root='./data/mnist', train=False, download=True, transform=transform)
testloader = torch.utils.data.DataLoader(
testset, batch_size=batch_size, shuffle=False,
num_workers=num_workers, pin_memory=pin_memory,
)
self.trainloader = trainloader
self.testloader = testloader
self.num_classes = 10