图片数据增强

import random
from albumentations.augmentations.transforms import Solarize
from albumentations.augmentations.transforms import ChannelShuffle
from albumentations.augmentations.transforms import InvertImg

iu=["1","2","3","4","5","6","7","8"]
iao=random.choice(iu)
print(iao)
input_img = imageio.imread('/media/lenovo/新加卷/PointCloudUDA/input/PnpAda_release_data/ct_train/img/0004.png')

def aug(input_img,iao):
    if iao==1:
        noise=iaa.AdditiveGaussianNoise(10,20)
        aug_img=noise.augment_image(input_img)
    if iao==7:
        noise=iaa.AdditiveGaussianNoise(10,30)
        aug_img=noise.augment_image(input_img)
    if iao==8:
        noise=iaa.AdditiveGaussianNoise(10,40)
        aug_img=noise.augment_image(input_img)
    if iao==2:
        contrast=iaa.GammaContrast((0.5, 2.0))
        aug_img = contrast.augment_image(input_img)
    if iao==3:
        contrast_sig = iaa.SigmoidContrast(gain=(5, 10), cutoff=(0.4, 0.6))
        aug_img = contrast_sig.augment_image(input_img)
    if iao == 4:
        contrast_lin = iaa.LinearContrast((0.6, 0.4))
        aug_img = contrast_lin.augment_image(input_img)
    if iao == 5:
        transform = ChannelShuffle(p=1.0)
        random.seed(7)
        aug_img = transform(image=input_img)['image']
    if iao == 6:
        transform = Solarize(threshold=200,  p=1.0)
        aug_img = transform(image=input_img)['image']
    return aug_img

以上代码仅涉及图片对比度、曝光度等，不涉及图片大小变化、翻转等。