Pytorch:torchvision.transforms图像增强方法和参数介绍

最近在使用pytorch构建网络,用到了torchvision.transforms图像增强方法。博主也在网上查找该方法的介绍,也有博主总结得很好,但只是介绍了方法名称,对方法具体的效果没有做详细介绍。本文将对各方法的参数做详细介绍,以求对方法的效果有更直观的印象。

另外,博主第一次写博客,本文将随着博主的学习推进持续更新,希望大家喜欢~

 

1. 仿射变换:transforms.RandomAffine

功能:随机仿射变换(包括随机旋转、随机左右移)

参数:

degrees:旋转角度;比如,degrees=30表示图像进行-30°~30°的随机旋转,等价于degrees=(-30, 30),效果如下图;

Pytorch:torchvision.transforms图像增强方法和参数介绍_第1张图片

translate:平移尺度;比如,translate=(0.3, 0.2)表示水平左移或右移图像宽度的0~0.3倍,垂直上移或下移图像高度的0~0.2倍,效果如下图;

Pytorch:torchvision.transforms图像增强方法和参数介绍_第2张图片

 

2. 修改亮度、对比度、饱和度、色调:transforms.ColorJitter

功能:随机修改亮度、对比度、饱和度、色调

参数:

brightness:亮度因子,调整亮度变暗或变亮的尺度。比如,brightness=0.3表示调整后的亮度为原来的0.7~1.3倍(该方法会随机选取某一倍数);brightness=(0.7, 1.3)等价于brightness=0.3;参数非负;

contrast:对比度因子,调整对比度的尺度;参数设置同brightness;

saturation:饱和度因子,调整饱和度的尺度;参数设置同brightness;

hue:色调因子,调整色调的尺度。参数设置与前三个不同,比如,hue=0.3表示调整后的色调为-0.3~0.3,并且hue\leq0.5;hue=(-0.3, 0.3)等价于hue=0.3。

 

比较忙,今天先写一点,后续会更新其他方法,大家海涵~

溜了溜了~

你可能感兴趣的:(python)