torchvision

Torchvision的使用

      • GaussianBlur(kernel[,sigma])

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结果:
torchvision_第1张图片torchvision_第2张图片

你可能感兴趣的:(torchvision,人工智能,python,pytorch,图像处理)