图像去抖动（高斯滤波、均值滤波、中值滤波）

最新推荐文章于 2024-07-14 23:52:32 发布

liujun_Pleasure

最新推荐文章于 2024-07-14 23:52:32 发布

阅读量316

点赞数 6

分类专栏：图像处理文章标签： python 算法人工智能

本文链接：https://blog.csdn.net/liujun221/article/details/136406524

版权

图像处理专栏收录该内容

9 篇文章 0 订阅

订阅专栏

采用高斯滤波、均值滤波、中值滤波进行组合去抖动操作

from PIL import Image, ImageFilter
import os

def remove_strong_shake(input_path, output_path, filter_type='gaussian', radius=5):
    # 打开图像
    img = Image.open(input_path)

    # 选择滤波器类型
    if filter_type == 'gaussian':
        # 使用更大半径的高斯模糊
        img_sharp = img.filter(ImageFilter.UnsharpMask(radius=radius, percent=200, threshold=1))
    elif filter_type == 'mean':
        # 使用均值模糊
        img_sharp = img.filter(ImageFilter.BLUR)
    elif filter_type == 'median':
        # 使用中值滤波
        img_sharp = img.filter(ImageFilter.MedianFilter(size=radius))

    # 保存去抖动后的图像
    img_sharp.save(output_path)

def batch_remove_shake(input_folder, output_folder, filter_type='gaussian', radius=5):
    # 确保输出文件夹存在，如果不存在则创建
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    # 遍历输入文件夹中的所有图像文件
    for filename in os.listdir(input_folder):
        if filename.endswith('.png') or filename.endswith('.jpg') or filename.endswith('.jpeg'):
            # 构建完整的文件路径
            input_path = os.path.join(input_folder, filename)

            # 构建输出文件路径
            output_path = os.path.join(output_folder, filename)

            # 调用去抖动函数
            remove_strong_shake(input_path, output_path, filter_type=filter_type, radius=radius)

# 输入文件夹路径和输出文件夹路径
input_folder = '1'
output_folder = '2'

# 调用函数进行批量去抖动
batch_remove_shake(input_folder, output_folder, filter_type='gaussian', radius=10)

print("去抖动完成！")