python jitter,Python transforms.ColorJitter方法代碼示例

本文整理匯總了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__