image_multithreads.py 文件源码

def data_augmentation(self, images, labels,
                          mode='train', 
                          flip=False, 
                          crop=False, padding=20, 
                          whiten=False, 
                          noise=False, noise_mean=0, noise_std=0.01,
                          resize=False, jitter=0.2):
        # ??????
        if resize:
            images, labels = self.image_resize(images, labels, jitter=jitter, mode=mode)
        # ????
        if crop:
            images = self.image_crop(images, padding=padding)
        # ????
        if flip:
            images, labels = self.image_flip(images, labels)
        # ????
        if whiten:
            images = self.image_whitening(images)
        # ????
        if noise:
            images = self.image_noise(images, mean=noise_mean, std=noise_std)

        return numpy.array(images, dtype='uint8'), numpy.array(labels, dtype='float32')