水平翻转 垂直翻转 颜色处理
1、随机水平翻转
官方文档 : 点击跳转
torchvision.transforms.RandomHorizontalFlip(p=0.5)
参数 :
-
p : 概率值,默认为 0.5。 图像会按照指定的概率随机做水平翻转
举例 :
from PIL import Image
import torchvision.transforms as transforms
import matplotlib.pyplot as pltoriginal_image = Image.open("image.jpg")
img = transforms.RandomHorizontalFlip(p=0.9)(original_image)plt.subplot(121)
plt.imshow(original_image)
plt.axis("off")plt.subplot(122)
plt.imshow(img)
plt.axis("off")plt.show()
2、随机垂直翻转
官方文档 : 点击跳转
torchvision.transforms.RandomVerticalFlip(p=0.5)
参数 :
-
p : 概率值,默认为 0.5。 图像会按照指定的概率随机做垂直翻转
举例 :
from PIL import Image
import torchvision.transforms as transforms
import matplotlib.pyplot as pltoriginal_image = Image.open("image.jpg")
img = transforms.RandomVerticalFlip(p=0.9)(original_image)plt.subplot(121)
plt.imshow(original_image)
plt.axis("off")plt.subplot(122)
plt.imshow(img)
plt.axis("off")plt.show()
3、图像颜色处理
官方文档 : 点击跳转
color_jitter = transforms. ColorJitter(brightness=0,contrast=0, saturation=0, hue=0)
参数:
-
brightness:亮度调整系数。 调整范围为 [ 1-brightness, 1+brightness ],默认值为 0。
-
contrast:对比度调整系数。 调整范国为 [ 1- contrast, 1+contrast ],默认值为 0。
-
saturation:饱和度调整系数。 调整范围为 [ 1- saturation, 1+ saturation ],默认值为 0。
-
hue:色调调整系数。 调整范围为 [ -hue, hue ],默认值为 0。