本文整理匯總了Python中torchvision.transforms.ColorJitter方法的典型用法代碼示例。如果您正苦於以下問題:Python transforms.ColorJitter方法的具體用法?Python transforms.ColorJitter怎麽用?Python transforms.ColorJitter使用的例子?那麽恭喜您, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在模塊torchvision.transforms的用法示例。
在下文中一共展示了transforms.ColorJitter方法的29個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於我們的係統推薦出更棒的Python代碼示例。
示例1: get_transforms
點讚 6
# 需要導入模塊: from torchvision import transforms [as 別名]
# 或者: from torchvision.transforms import ColorJitter [as 別名]
def get_transforms(self):
if self.config['train']['transform'] == True:
self.train_transform = transforms.Compose([
transforms.ColorJitter(brightness=self.config['augmentation']['brightness'], contrast=self.config['augmentation']['contrast'], saturation=self.config['augmentation']['saturation'], hue=self.config['augmentation']['hue']),
transforms.ToTensor(),
])
else:
self.train_transform = transforms.Compose([
transforms.ToTensor(),
])
self.test_transform = transforms.Compose([
transforms.ToTensor(),
])
self.target_transform = transforms.Compose([
transforms.ToTensor(),
])
#Does data augmentation, ie. tranforms images by changing colour, hue brightness, etc., and returns tensor
開發者ID:mayank-git-hub,項目名稱:Text-Recognition,代碼行數:23,
示例2: cifar10_train_transform
點讚 6
# 需要導入模塊: from torchvision import transforms [as 別名]
# 或者: from torchvision.transforms import ColorJitter [as 別名]
def cifar10_train_transform(ds_metainfo,
mean_rgb=(0.4914, 0.4822, 0.4465),
std_rgb=(0.2023, 0.1994, 0.2010),
jitter_param=0.4):
assert (ds_metainfo is not None)
assert (ds_metainfo.input_image_size[0] == 32)
return transforms.Compose([
transforms.RandomCrop(
size=32,
padding=4),
transforms.RandomHorizontalFlip(),
transforms.ColorJitter(
brightness=jitter_param,
contrast=jitter_param,
saturation=jitter_param),
transforms.ToTensor(),
transforms.Normalize(
mean=mean_rgb,
std=std_rgb)
])
開發者ID:osmr,項目名稱:imgclsmob,代碼行數:22,
示例3: get_jig_train_transformers
點讚 6
# 需要導入模塊: from torchvision import transforms [as 別名]
# 或者: from torchvision.transforms import ColorJitter [as 別名]
def get_jig_train_transformers(args):
size = args.img_transform.random_resize_crop.size
scale = args.img_transform.random_resize_crop.scale
img_tr = [transforms.RandomResizedCrop((int(size[0]), int(size[1])), (scale[0], scale[1]))]
if args.img_transform.random_horiz_flip > 0.0:
img_tr.append(transforms.RandomHorizontalFlip(args.img_transform.random_horiz_flip))
if args.img_transform.jitter > 0.0:
img_tr.append(transforms.ColorJitter(
brightness=args.img_transform.jitter, contrast=args.img_transform.jitter,
saturation=args.jitter, hue=min(0.5, args.jitter)))
tile_tr = []
if args.jig_transform.tile_random_grayscale:
tile_tr.append(transforms.RandomGrayscale(args.jig_transform.tile_random_grayscale))
mean = args.normalize.mean
std = args.normalize.std
tile_tr = tile_tr + [transforms.ToTensor(), transforms.Normalize(mean=mean, std=std)]
return transforms.Compose(img_tr), transforms.Compose(tile_tr)
開發者ID:Jiaolong,項目名稱:self-supervised-da,代碼行數:21,
示例4: get_rot_train_transformers
點讚 6
# 需要導入模塊: from torchvision import transforms [as 別名]
# 或者: from torchvision.transforms import ColorJitter [as 別名]
def get_rot_train_transformers(args):
size = args.img_transform.random_resize_crop.size
scale = args.img_transform.random_resize_crop.scale
img_tr = [transforms.RandomResizedCrop((int(size[0]), int(size[1])), (scale[0], scale[1]))]
if args.img_transform.random_horiz_flip > 0.0:
img_tr.append(transforms.RandomHorizontalFlip(args.img_transform.random_horiz_flip))
if args.img_transform.jitter > 0.0:
img_tr.append(transforms.ColorJitter(
brightness=args.img_transform.jitter, contrast=args.img_transform.jitter,
saturation=args.jitter, hue=min(0.5, args.jitter)))
mean = args.normalize.mean
std = args.normalize.std
img_tr += [transforms.ToTensor(), transforms.Normalize(mean=mean, std=std)]
return transforms.Compose(img_tr)
開發者ID:Jiaolong,項目名稱:self-supervised-da,代碼行數:18,
示例5: preprocess
點讚 6
# 需要導入模塊: from torchvision import transforms [as 別名]
# 或者: from torchvision.transforms import ColorJitter [as 別名]
def preprocess(self):
if self.train:
return transforms.Compose([
transforms.RandomResizedCrop(self.image_size),
transforms.RandomHorizontalFlip(),
transforms.ColorJitter(brightness=0.4, contrast=0.4, saturation=0.4, hue=0.2),
transforms.ToTensor(),
transforms.Normalize(self.mean, self.std),
])
else:
return transforms.Compose([
transforms.Resize((int(self.image_size / 0.875), int(self.image_size / 0.875))),
transforms.CenterCrop(self.image_size),
transforms.ToTensor(),
transforms.Normalize(self.mean, self.std),
])
開發者ID:wandering007,項目名稱:nasnet-pytorch,代碼行數:18,
示例6: __init__
點讚 6
# 需要導入模塊: from torchvision import transforms [as 別名]
# 或者: from torchvision.transforms import ColorJitter [as 別名]
def __init__(self, root_dir, txtlist, use_noise, length):
self.path = []
self.real_path = []
self.use_noise = use_noise
self.root = root_dir
input_file = open(txtlist)
while 1:
input_line = input_file.readline()
if not input_line:
break
if input_line[-1:] == '\n':
input_line = input_line[:-1]
self.path.append(copy.deepcopy(input_line))
if input_line[:5] == 'data/':
self.real_path.append(copy.deepcopy(input_line))
input_file.close()
self.length = length
self.data_len = len(self.path)
self.back_len = len(self.real_path)
self.trancolor = transforms.ColorJitter(0.2, 0.2, 0.2, 0.05)
self.norm = transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
self.back_front = np.array([[1 for i in range(640)] for j in range(480)])
開發者ID:j96w,項目名稱:DenseFusion,代碼行數:26,
示例7: __init__