Python实现图像去噪方式(中值去噪和均值去噪)


Posted in Python onDecember 18, 2019

实现对图像进行简单的高斯去噪和椒盐去噪。

代码如下:

import numpy as np
from PIL import Image
import matplotlib.pyplot as plt
import random
import scipy.misc
import scipy.signal
import scipy.ndimage
from matplotlib.font_manager import FontProperties
font_set = FontProperties(fname=r"c:\windows\fonts\simsun.ttc", size=10)
 
def medium_filter(im, x, y, step):
  sum_s = []
  for k in range(-int(step / 2), int(step / 2) + 1):
    for m in range(-int(step / 2), int(step / 2) + 1):
      sum_s.append(im[x + k][y + m])
  sum_s.sort()
  return sum_s[(int(step * step / 2) + 1)]
 
 
def mean_filter(im, x, y, step):
  sum_s = 0
  for k in range(-int(step / 2), int(step / 2) + 1):
    for m in range(-int(step / 2), int(step / 2) + 1):
      sum_s += im[x + k][y + m] / (step * step)
  return sum_s
 
 
def convert_2d(r):
  n = 3
  # 3*3 滤波器, 每个系数都是 1/9
  window = np.ones((n, n)) / n ** 2
  # 使用滤波器卷积图像
  # mode = same 表示输出尺寸等于输入尺寸
  # boundary 表示采用对称边界条件处理图像边缘
  s = scipy.signal.convolve2d(r, window, mode='same', boundary='symm')
  return s.astype(np.uint8)
 
 
def convert_3d(r):
  s_dsplit = []
  for d in range(r.shape[2]):
    rr = r[:, :, d]
    ss = convert_2d(rr)
    s_dsplit.append(ss)
  s = np.dstack(s_dsplit)
  return s
 
 
def add_salt_noise(img):
  rows, cols, dims = img.shape
  R = np.mat(img[:, :, 0])
  G = np.mat(img[:, :, 1])
  B = np.mat(img[:, :, 2])
 
  Grey_sp = R * 0.299 + G * 0.587 + B * 0.114
  Grey_gs = R * 0.299 + G * 0.587 + B * 0.114
 
  snr = 0.9
 
  noise_num = int((1 - snr) * rows * cols)
 
  for i in range(noise_num):
    rand_x = random.randint(0, rows - 1)
    rand_y = random.randint(0, cols - 1)
    if random.randint(0, 1) == 0:
      Grey_sp[rand_x, rand_y] = 0
    else:
      Grey_sp[rand_x, rand_y] = 255
  #给图像加入高斯噪声
  Grey_gs = Grey_gs + np.random.normal(0, 48, Grey_gs.shape)
  Grey_gs = Grey_gs - np.full(Grey_gs.shape, np.min(Grey_gs))
  Grey_gs = Grey_gs * 255 / np.max(Grey_gs)
  Grey_gs = Grey_gs.astype(np.uint8)
 
  # 中值滤波
  Grey_sp_mf = scipy.ndimage.median_filter(Grey_sp, (7, 7))
  Grey_gs_mf = scipy.ndimage.median_filter(Grey_gs, (8, 8))
 
  # 均值滤波
  Grey_sp_me = convert_2d(Grey_sp)
  Grey_gs_me = convert_2d(Grey_gs)
 
  plt.subplot(321)
  plt.title('加入椒盐噪声',fontproperties=font_set)
  plt.imshow(Grey_sp, cmap='gray')
  plt.subplot(322)
  plt.title('加入高斯噪声',fontproperties=font_set)
  plt.imshow(Grey_gs, cmap='gray')
 
  plt.subplot(323)
  plt.title('中值滤波去椒盐噪声(8*8)',fontproperties=font_set)
  plt.imshow(Grey_sp_mf, cmap='gray')
  plt.subplot(324)
  plt.title('中值滤波去高斯噪声(8*8)',fontproperties=font_set)
  plt.imshow(Grey_gs_mf, cmap='gray')
 
  plt.subplot(325)
  plt.title('均值滤波去椒盐噪声',fontproperties=font_set)
  plt.imshow(Grey_sp_me, cmap='gray')
  plt.subplot(326)
  plt.title('均值滤波去高斯噪声',fontproperties=font_set)
  plt.imshow(Grey_gs_me, cmap='gray')
  plt.show()
 
 
def main():
  img = np.array(Image.open('E:/pycharm/GraduationDesign/Test/testthree.png'))
  add_salt_noise(img)
 
 
if __name__ == '__main__':
  main()

效果如下

Python实现图像去噪方式(中值去噪和均值去噪)

以上这篇Python实现图像去噪方式(中值去噪和均值去噪)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
几个提升Python运行效率的方法之间的对比
Apr 03 Python
用Python进行TCP网络编程的教程
Apr 29 Python
python+django快速实现文件上传
Oct 24 Python
python使用xlrd与xlwt对excel的读写和格式设定
Jan 21 Python
Python排序搜索基本算法之希尔排序实例分析
Dec 09 Python
将Pytorch模型从CPU转换成GPU的实现方法
Aug 19 Python
解决pyshp UnicodeDecodeError的问题
Dec 06 Python
Python 3 使用Pillow生成漂亮的分形树图片
Dec 24 Python
tensorflow2.0与tensorflow1.0的性能区别介绍
Feb 07 Python
opencv python 图片读取与显示图片窗口未响应问题的解决
Apr 24 Python
python使用numpy中的size()函数实例用法详解
Jan 29 Python
Pytorch 中net.train 和 net.eval的使用说明
May 22 Python
python 中值滤波,椒盐去噪,图片增强实例
Dec 18 #Python
Django中使用MySQL5.5的教程
Dec 18 #Python
Python hashlib加密模块常用方法解析
Dec 18 #Python
Python实现中值滤波去噪方式
Dec 18 #Python
详解Python中字符串前“b”,“r”,“u”,“f”的作用
Dec 18 #Python
Python字典底层实现原理详解
Dec 18 #Python
Python利用PyExecJS库执行JS函数的案例分析
Dec 18 #Python
You might like
thinkphp3.x连接mysql数据库的方法(具体操作步骤)
2016/05/19 PHP
php用户密码加密算法分析【Discuz加密算法】
2016/10/12 PHP
浅谈php://filter的妙用
2019/03/05 PHP
PHP从零开始打造自己的MVC框架之入口文件实现方法详解
2019/06/03 PHP
Extjs 继承Ext.data.Store不起作用原因分析及解决
2013/04/15 Javascript
JS获取数组最大值、最小值及长度的方法
2015/11/24 Javascript
JS中使用apply方法通过不同数量的参数调用函数的方法
2016/05/31 Javascript
微信小程序 在Chrome浏览器上运行以及WebStorm的使用
2016/09/27 Javascript
AngularJS 霸道的过滤器小结
2017/04/26 Javascript
利用prop-types第三方库对组件的props中的变量进行类型检测
2017/05/02 Javascript
基于ES6 Array.of的用法(实例讲解)
2017/09/05 Javascript
微信小程序实现发送验证码按钮效果
2018/12/20 Javascript
微信小程序个人中心的列表控件实现代码
2020/04/26 Javascript
Python编程实现粒子群算法(PSO)详解
2017/11/13 Python
关于Python的一些学习总结
2018/05/25 Python
Selenium定时刷新网页的实现代码
2018/10/31 Python
python写日志文件操作类与应用示例
2019/07/01 Python
在linux下实现 python 监控usb设备信号
2019/07/03 Python
python实现单链表的方法示例
2019/09/03 Python
python使用socket实现的传输demo示例【基于TCP协议】
2019/09/24 Python
python itsdangerous模块的具体使用方法
2020/02/17 Python
在django admin详情表单显示中添加自定义控件的实现
2020/03/11 Python
Python基于read(size)方法读取超大文件
2020/03/12 Python
python使用隐式循环快速求和的实现示例
2020/09/11 Python
Django正则URL匹配实现流程解析
2020/11/13 Python
Desigual英国官网:在线购买原创服装
2018/03/09 全球购物
能源工程专业应届生求职信
2014/03/01 职场文书
党员承诺书怎么写
2014/05/20 职场文书
大学计划书范文800字
2014/08/14 职场文书
就业协议书
2014/09/12 职场文书
党的群众路线教育实践活动心得体会(乡镇)
2014/11/03 职场文书
2014年依法行政工作总结
2014/11/19 职场文书
2014年幼儿园教学工作总结
2014/12/04 职场文书
铁路安全反思材料
2014/12/24 职场文书
法律讲堂观后感
2015/06/11 职场文书
深入理解mysql事务隔离级别和存储引擎
2022/04/12 MySQL