python 图像去噪

使用python对图像进行去噪,较为简单,为追求更好的效果,可能还需要上模型。

效果对比

假设我关心安全帽部位,可以看到右侧去噪之后,安全帽轮廓更加干净。

我的模型是用较为清晰的图像训练的。在我的模型上,左侧这种较为模糊的图片图片会发生误识别(即将图中人员识别为未戴安全帽),右侧会识别为"带了安全帽"。

代码

python 复制代码
import cv2
import os
from PIL import Image
import numpy as np

def denoise_image(image_path, output_path):
    # 使用 OpenCV 读取图像
    img = cv2.imread(image_path)
    
    if img is None:
        print(f"Error loading image {image_path}")
        return

    # 去噪处理
    # 适用于彩色图像的去噪
    denoised_img = cv2.fastNlMeansDenoisingColored(img, None, 4, 4, 7, 21)
    
    # 将处理后的图像转换为 PIL 格式并保存
    denoised_pil_img = Image.fromarray(cv2.cvtColor(denoised_img, cv2.COLOR_BGR2RGB))
    denoised_pil_img.save(output_path)
    print(f"Saved denoised image to {output_path}")

def process_images_in_folder(folder_path, output_folder):
    # 确保输出文件夹存在
    os.makedirs(output_folder, exist_ok=True)
    
    # 遍历文件夹中的所有图像文件
    for file_name in os.listdir(folder_path):
        if file_name.lower().endswith(('.png', '.jpg', '.jpeg')):
            input_path = os.path.join(folder_path, file_name)
            base_name = os.path.splitext(file_name)[0]
            output_file_name = f"{base_name}_denoise_img{os.path.splitext(file_name)[1]}"
            output_path = os.path.join(output_folder, output_file_name)
            denoise_image(input_path, output_path)

# 替换以下路径
input_folder_path = r'E:\noise'
output_folder_path = r'E:\denoise'


# 执行处理
process_images_in_folder(input_folder_path, output_folder_path)
相关推荐
会的全对٩(ˊᗜˋ*)و8 分钟前
【数据挖掘】数据挖掘综合案例—银行精准营销
人工智能·经验分享·python·数据挖掘
___波子 Pro Max.27 分钟前
GitHub Actions配置python flake8和black
python·black·flake8
rzl0244 分钟前
java web5(黑马)
java·开发语言·前端
时序数据说1 小时前
为什么时序数据库IoTDB选择Java作为开发语言
java·大数据·开发语言·数据库·物联网·时序数据库·iotdb
jingling5551 小时前
面试版-前端开发核心知识
开发语言·前端·javascript·vue.js·面试·前端框架
阿蒙Amon1 小时前
【Python小工具】使用 OpenCV 获取视频时长的详细指南
python·opencv·音视频
m0_687399841 小时前
写一个Ununtu C++ 程序,调用ffmpeg API, 来判断一个数字电影的视频文件mxf 是不是Jpeg2000?
开发语言·c++·ffmpeg
爱上语文1 小时前
Redis基础(5):Redis的Java客户端
java·开发语言·数据库·redis·后端
A~taoker2 小时前
taoker的项目维护(ng服务器)
java·开发语言
萧曵 丶2 小时前
Rust 中的返回类型
开发语言·后端·rust