transform:数据预处理代码，可以自定义数据增广等

最新推荐文章于 2024-06-13 10:35:50 发布

Jane_JU

最新推荐文章于 2024-06-13 10:35:50 发布

阅读量1.5k

点赞数

分类专栏：行人重识别文章标签：计算机视觉人工智能

本文链接：https://blog.csdn.net/Jane_JU/article/details/122280066

版权

行人重识别专栏收录该内容

6 篇文章 0 订阅

订阅专栏

from __future__ import absolute_import

from torchvision.transforms import *
from PIL import Image
import random
import numpy as np

class Random2DTranslation(object):
    """
    With a probability, first increase image size to (1 + 1/8), and then perform random crop.

    Args:
        height (int): target height.
        width (int): target width.
        p (float): probability of performing this transformation. Default: 0.5.
    """
    def __init__(self, height, width, p=0.5, interpolation=Image.BILINEAR):
        self.height = height
        self.width = width
        self.p = p
        self.interpolation = interpolation

    def __call__(self, img):
        """
        Args:
            img (PIL Image): Image to be cropped.

        Returns:
            PIL Image: Cropped image.
        """
        if random.random() < self.p:
            return img.resize((self.width, self.height), self.interpolation)
        new_width, new_height = int(round(self.width * 1.125)), int(round(self.height * 1.125))
        resized_img = img.resize((new_width, new_height), self.interpolation)
        x_maxrange = new_width - self.width
        y_maxrange = new_height - self.height
        x1 = int(round(random.uniform(0, x_maxrange)))
        y1 = int(round(random.uniform(0, y_maxrange)))
        croped_img = resized_img.crop((x1, y1, x1 + self.width, y1 + self.height))
        return croped_img
if __name__ == '__main__':
    img = Image.open('/home/ubuntu/reid/AlignedReID/data/market1501/bounding_box_train/0002_c1s1_000451_03.jpg')
    transform = Random2DTranslation(256, 128, 0.5)
    img_t = transform(img)
    import matplotlib.pyplot as plt
    
    plt.figure(12)
    plt.subplot(121)
    plt.imshow(img)
    plt.subplot(122)
    plt.imshow(img_t)
    plt.show()

在这里插入图片描述

if __name__ == '__main__':
    img = Image.open('/home/ubuntu/reid/AlignedReID/data/market1501/bounding_box_train/0002_c1s1_000451_03.jpg')
    transform = transforms.Compose(
        [
            Random2DTranslation(256, 128, 0.5),
            transforms.RandomHorizontalFlip(),
            #transforms.ToTensor(),
        ]
    )
    img_t = transform(img)
    import matplotlib.pyplot as plt
    
    plt.figure(12)
    plt.subplot(121)
    plt.imshow(img)
    plt.subplot(122)
    plt.imshow(img_t)
    plt.show()