python crop_Python transforms.RandomCrop方法代码示例

本文整理汇总了Python中torchvision.transforms.RandomCrop方法的典型用法代码示例。如果您正苦于以下问题：Python transforms.RandomCrop方法的具体用法？Python transforms.RandomCrop怎么用？Python transforms.RandomCrop使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在模块torchvision.transforms的用法示例。

在下文中一共展示了transforms.RandomCrop方法的26个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于我们的系统推荐出更棒的Python代码示例。

示例1: load_data

​点赞 6

​

# 需要导入模块: from torchvision import transforms [as 别名]

# 或者: from torchvision.transforms import RandomCrop [as 别名]

def load_data(data_folder, batch_size, phase='train', train_val_split=True, train_ratio=.8):

transform_dict = {

'train': transforms.Compose(

[transforms.Resize(256),

transforms.RandomCrop(224),

transforms.RandomHorizontalFlip(),

transforms.ToTensor(),

transforms.Normalize(mean=[0.485, 0.456, 0.406],

std=[0.229, 0.224, 0.225]),

]),

'test': transforms.Compose(

[transforms.Resize(224),

transforms.ToTensor(),

transforms.Normalize(mean=[0.485, 0.456, 0.406],

std=[0.229, 0.224, 0.225]),

])}

data = datasets.ImageFolder(root=data_folder, transform=transform_dict[phase])

if phase == 'train':

if train_val_split:

train_size = int(train_ratio * len(data))

test_size = len(data) - train_size

data_train, data_val = torch.utils.data.random_split(data, [train_size, test_size])

train_loader = torch.utils.data.DataLoader(data_train, batch_size=batch_size, shuffle=True, drop_last=True,

num_workers=4)

val_loader = torch.utils.data.DataLoader(data_val, batch_size=batch_size, shuffle=False, drop_last=False,

num_workers=4)

return [train_loader, val_loader]

else:

train_loader = torch.utils.data.DataLoader(data, batch_size=batch_size, shuffle=True, drop_last=True,

num_workers=4)

return train_loader

else:

test_loader = torch.utils.data.DataLoader(data, batch_size=batch_size, shuffle=False, drop_last=False,

num_workers=4)

return test_loader

## Below are for ImageCLEF datasets

开发者ID:jindongwang，项目名称:transferlearning，代码行数:40，

示例2: load_imageclef_train

​点赞 6

​

# 需要导入模块: from torchvision import transforms [as 别名]

# 或者: from torchvision.transforms import RandomCrop [as 别名]

def load_imageclef_train(root_path, domain, batch_size, phase):

transform_dict = {

'src': transforms.Compose(

[transforms.Resize((256, 256)),

transforms.RandomCrop(224),

transforms.RandomHorizontalFlip(),

transforms.ToTensor(),

transforms.Normalize(mean=[0.485, 0.456, 0.406],

std=[0.229, 0.224, 0.225]),

]),

'tar': transforms.Compose(

[transforms.Resize((224, 224)),

transforms.ToTensor(),

transforms.Normalize(mean=[0.485, 0.456, 0.406],

std=[0.229, 0.224, 0.225]),

])}

data = ImageCLEF(root_dir=root_path, domain=domain, transform=transform_dict[phase])

train_size = int(0.8 * len(data))

test_size = len(data) - train_size

data_train, data_val = torch.utils.data.random_split(data, [train_size, test_size])

train_loader = torch.utils.data.DataLoader(data_train, batch_size=batch_size, shuffle=True, drop_last=False,

num_workers=4)

val_loader = torch.utils.data.DataLoader(data_val, batch_size=batch_size, shuffle=True, drop_last=False,

num_workers=4)

return train_loader, val_loader

开发者ID:jindongwang，项目名称:transferlearning，代码行数:27，

示例3: load_imageclef_test

​点赞 6

​

# 需要导入模块: from torchvision import transforms [as 别名]

# 或者: from torchvision.transforms import RandomCrop [as 别名]

def load_imageclef_test(root_path, domain, batch_size, phase):

transform_dict = {

'src': transforms.Compose(

[transforms.Resize((256,256)),

transforms.RandomCrop(224),

transforms.RandomHorizontalFlip(),

transforms.ToTensor(),

transforms.Normalize(mean=[0.485, 0.456, 0.406],

std=[0.229, 0.224, 0.225]),

]),

'tar': transforms.Compose(

[transforms.Resize((224, 224)),

transforms.ToTensor(),

transforms.Normalize(mean=[0.485, 0.456, 0.406],

std=[0.229, 0.224, 0.225]),

])}

data = ImageCLEF(root_dir=root_path, domain=domain, transform=transform_dict[phase])

data_loader = torch.utils.data.DataLoader(data, batch_size=batch_size, shuffle=True, drop_last=False, num_workers=4)

return data_loader

开发者ID:jindongwang，项目名称:transferlearning，代码行数:21，

示例4: load_training

​点赞 6

​

# 需要导入模块: from torchvision import transforms [as 别名]

# 或者: from torchvision.transforms import RandomCrop [as 别名]

def load_training(root_path, dir, batch_size, kwargs):

transform = transforms.Compose(

[transforms.Resize([256, 256]),

transforms.RandomCrop(224),

transforms.RandomHorizontalFlip(),

transforms.ToTensor()])

data = datasets.ImageFolder(root=root_path + dir, transform=transform)

train_loader = torch.utils.data.DataLoader(data, batch_size=batch_size, shuffle=True, drop_last=True, **kwargs)

return train_loader

开发者ID:jindongwang，项目名称:transferlearning，代码行数:12，

示例5: main

​点赞 6

​

# 需要导入模块: from torchvision import transforms [as 别名]

# 或者: from torchvision.transforms import RandomCrop [as 别名]

def main():

best_acc = 0

device = 'cuda' if torch.cuda.is_available() else 'cpu'

print('==> Preparing data..')

transforms_train = transforms.Compose([

transforms.RandomCrop(32, padding=4),

transforms.RandomHorizontalFlip(),

transforms.ToTensor(),