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;形状与输入保持一致。