mmdetection阅读记录-imagecorruptions模块

imagecorruptions安装

国内使用清华源，终端输入：

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple imagecorruptions

imagecorruptions使用说明

imagecorruptions模块提供了多种为图片增加噪声的方法。
在mmdetection中用到了imagecorruptions模块中的corrupt函数：

def corrupt(image, severity=1, corruption_name=None, corruption_number=-1):

参数：

image(numpy.ndarray):
需要添加噪声的图片输入，要求是numpy array类型，取值范围在[0,255]，数据类型要求为np.uint8，输入形状为（HxWxC）或（HxW），且H和W至少为32像素，通道数C为1或3；

severity(int):
对图片添加噪声的程度，取值为1-5之间的整数。

corruption_name(str):
指定噪声函数，必须为以下之一：

‘gaussian_noise’, ‘shot_noise’, ‘impulse_noise’, ‘defocus_blur’,
‘glass_blur’, ‘motion_blur’, ‘zoom_blur’, ‘snow’, ‘frost’, ‘fog’,
‘brightness’, ‘contrast’, ‘elastic_transform’, ‘pixelate’, ‘jpeg_compression’,
‘speckle_noise’, ‘gaussian_blur’, ‘spatter’, ‘saturate’;

corruption_number(int):
corruption_name在上述列表中的位置，必须是0-18之间的整数；设置该参数便于循环中的使用；

Return:
返回添加过噪声的图片，数据类型为numpy.ndarray；形状与输入保持一致。