transforms.RandomSolarize(threshold, p=0.5)
通过反转阈值以上的所有像素值,以给定的概率随机对图像进行日光化
import torchvision.transforms as transform
from PIL import Image
import matplotlib.pyplot as plt
import numpy as np
import torch
img0=Image.open('lin-xiao-xun-000003.jpg')
img1=transform.RandomInvert(p=0.8)(img0)
axs = plt.figure().subplots(1,2)
axs[0].imshow(img0);axs[0].set_title('src');axs[0].axis('off')
axs[1].imshow(img1);axs[1].set_title('RandomInvert');axs[1].axis('off')
plt.show()
实图演示: