Python如何去除图片干扰代码示例_Python

在python中去除图片干扰，需根据干扰类型（如噪声、特定物体、强光等）选择合适的方法。以下是分场景解决方案及代码示例：

一、噪声去除

1. 高斯噪声（像素值正态分布扰动）

方法：高斯滤波、双边滤波、小波变换

代码示例（opencv）：

import cv2
import numpy as np

# 读取图像并添加高斯噪声
image = cv2.imread('noisy_image.jpg')
noise = np.random.normal(0, 25, image.shape).astype(np.uint8)
noisy_image = cv2.add(image, noise)

# 高斯滤波去噪
gaussian_filtered = cv2.gaussianblur(noisy_image, (5, 5), 0)

# 双边滤波（保留边缘）
bilateral_filtered = cv2.bilateralfilter(noisy_image, d=9, sigmacolor=75, sigmaspace=75)

cv2.imshow('original', image)
cv2.imshow('gaussian filtered', gaussian_filtered)
cv2.imshow('bilateral filtered', bilateral_filtered)
cv2.waitkey(0)

2. 椒盐噪声（随机黑白像素点）

方法：中值滤波

代码示例（opencv）：

# 添加椒盐噪声（示例）
x = image.reshape(-1)
snr = 0.85
noise_num = int(x.size * (1 - snr))
random_indices = np.random.choice(x.size, noise_num, replace=false)
x[random_indices] = np.random.choice([0, 255], noise_num)
noisy_image = x.reshape(image.shape)

# 中值滤波去噪
median_filtered = cv2.medianblur(noisy_image, 5)

3. 复杂噪声（如伪影）

方法：非局部均值去噪（nlm）

代码示例（scikit-image）：

from skimage import io, img_as_float
from skimage.restoration import denoise_nl_means

image = img_as_float(io.imread('noisy_image.jpg'))
denoised = denoise_nl_means(image, h=0.1, fast_mode=true, patch_size=5, patch_distance=3)

二、特定干扰去除

1. 干扰线（如扫描文档中的横线）

方法：二值化 + 邻域分析

代码示例（pillow）：

from pil import image, imagefilter

def remove_lines(image_path, threshold=128):
    image = image.open(image_path).convert('l')  # 转为灰度
    binarized = image.point(lambda x: 0 if x < threshold else 255, '1')
    clean = binarized.copy()
    width, height = binarized.size

    for y in range(1, height-1):
        for x in range(1, width-1):
            if binarized.getpixel((x, y)) == 0:
                neighbors = [binarized.getpixel((x-1, y)), binarized.getpixel((x+1, y)),
                             binarized.getpixel((x, y-1)), binarized.getpixel((x, y+1))]
                if neighbors.count(0) >= 2:
                    clean.putpixel((x, y), 255)
    return clean

cleaned_image = remove_lines('document.jpg')
cleaned_image.save('cleaned_document.jpg')

2. 强光干扰（过曝区域）

方法：颜色空间转换 + 阈值调整

代码示例（opencv）：

import cv2
import numpy as np

image = cv2.imread('overexposed.jpg')
hsv = cv2.cvtcolor(image, cv2.color_bgr2hsv)
lower = np.array([0, 0, 200])  # v通道阈值
upper = np.array([180, 255, 255])
mask = cv2.inrange(hsv, lower, upper)

# 降低过曝区域亮度
image[mask != 0] = cv2.add(image[mask != 0], (0, 0, -80))
cv2.imwrite('corrected.jpg', image)

三、深度学习进阶方案

对于复杂场景（如混合噪声、纹理干扰），可使用预训练模型（如u-net、dncnn）：

import torch
from torchvision import models

# 加载预训练去噪模型（示例）
model = models.dncnn().eval()
model.load_state_dict(torch.load('dncnn_pretrained.pth'))

# 预处理输入
input_tensor = preprocess(noisy_image)  # 需自定义预处理函数
with torch.no_grad():
    output = model(input_tensor)
denoised_image = postprocess(output)  # 自定义后处理函数