Torchvision的使用
GaussianBlur(kernel[,sigma])
作用:对输入的图像进行高斯模糊
返回:PIL image or Tensor
例:
from PIL import Image
import numpy as np
import torch
import os
import sys
import glob
from torchvision.transforms import GaussianBlur
def gauss(image,save_path,name):
gauss=GaussianBlur(kernel_size=(5,9),sigma=(0.1,5))
iam=gauss(image)
path=os.path.join(save_path,name)
iam.save(path)
if __name__=="__main__":
src_path='./data/datasets/gts/face'
save_path='./data/datasets/gts/blurface'
names=os.listdir(src_path)
files=glob.glob(f"{src_path}/*.png")
for i,file in enumerate(files):
name=file.split("/")[-1]
image=Image.open(file)
gauss(image,save_path,name)
原图vs结果: