Python+opencv 均值滤波

图像处理 专栏收录该内容
8 篇文章 0 订阅
import cv2
import matplotlib.pyplot as plt
import matplotlib
import numpy as np
import random

# 设置画图支持中文
font = {'family': 'SimHei'}
matplotlib.rc('font', **font)


def add_noise(image, noise_num):
    '''
    :param image: 待添加噪声的图像
    :param noise_num: 需要添加的噪声数量
    :return: 添加噪声后的图像
    '''
    import copy
    img = copy.deepcopy(image)
    img_size = img.shape
    rows, cols = img_size[0], img_size[1]
    for i in range(noise_num):
        x = np.random.randint(0, rows)
        y = np.random.randint(0, cols)
        if len(img_size) > 2:   # if判断是彩图还是灰度图
            if random.random() > 0.5:
                img[x, y, :] = 255
            else:
                img[x, y, :] = 0
        else:
            if random.random() > 0.5:
                img[x, y] = 255
            else:
                img[x, y] = 0
    return img

# 图像路径
img_path = 'lenna.bmp'

# 按三通道图像格式读入
source = cv2.imread(img_path, cv2.IMREAD_UNCHANGED)
if len(source.shape) > 2:  # 彩图显示到plt上需要将BGR转为RGB
    source = cv2.cvtColor(source, cv2.COLOR_BGR2RGB)
# 对原图均值滤波,核大小为3x3
source_average_filtering_3x3 = cv2.blur(source, ksize=(3, 3))
# 对原图均值滤波,核大小为5x5
source_average_filtering_5x5 = cv2.blur(source, ksize=(5, 5))
# 对原图添加10000个噪声点
source_noise = add_noise(source, 10000)
# 对噪声图均值滤波,核大小为3x3
source_noise_average_filtering_3x3 = cv2.blur(source_noise, ksize=(3, 3))
# 对噪声图均值滤波,核大小为5x5
source_noise_average_filtering_5x5 = cv2.blur(source_noise, ksize=(5, 5))


imgs = [source,
        source_average_filtering_3x3,
        source_average_filtering_5x5,
        source_noise,
        source_noise_average_filtering_3x3,
        source_noise_average_filtering_5x5]
titles = ['原图',
          '原图均值滤波(3x3)',
          '原图均值滤波(5x5)',
          '噪声图',
          '噪声图均值滤波(3x3)',
          '噪声图均值滤波(5x5)']

for idx in range(len(imgs)):
    plt.subplot(2,3,idx+1)  # 2行3列,将图像放置在第idx+1位置上(从左往右,从上到下计数)
    plt.imshow(imgs[idx], cmap='gray')
    plt.title(titles[idx])

# 保存图像,bbox_inches='tight'去除边缘空白,dpi设置图像清晰度
plt.savefig('均值滤波.png', bbox_inches='tight', dpi=300)
plt.show()

在这里插入图片描述

  • 0
    点赞
  • 1
    评论
  • 2
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: 游动-白 设计师:白松林 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值