深度学习中RGB影像图的直方图均衡化python代码and对图片中指定部分做基于掩模的特定区域直方图均衡化

本文主要是介绍深度学习中RGB影像图的直方图均衡化python代码and对图片中指定部分做基于掩模的特定区域直方图均衡化,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

深度学习很重要的预处理步骤
就是需要对做直方图均衡化
其中主要分成灰度图以及RGB图的直方图均衡化
这俩的方法和代码不同
想要去看具体原理的朋友可以查看下面这篇博客的内容
写的很详细
颜色直方图均衡化(https://www.cnblogs.com/wancy/p/17668345.html)

我们这个场景中会用到的就是颜色直方图均衡化了
其中包含三种方法


方法1.在BGR颜色空间下进行直方图均衡化,可以分别对每个通道进行均衡化。

以下是批量读取某个文件夹中的所有图片,并对每张图片做RGB直方图均衡化,使用OpenCV库实现彩色图像直方图均衡化(在BGR颜色空间)的代码:

import cv2
import os
import shutilfilePath = r"F:\deepl\sample\complete\road3\white"  # 用于获取文件名称列表
new_path = r"F:\deepl\sample\complete\road3\white-rgb"  # 目标文件夹
#move_path = r"F:\deepl\sample\complete\water2\sat"  # 目标文件夹file_names = filter(lambda x: x.find('png')!=-1, os.listdir(filePath))# print(file_list)for file in file_names:
# 读取彩色图像path=filePath+'\\'+fileimg = cv2.imread(path)# 分离BGR图像的通道b, g, r = cv2.split(img)# 对每个通道进行直方图均衡化equ_b = cv2.equalizeHist(b)equ_g = cv2.equalizeHist(g)equ_r = cv2.equalizeHist(r)equ_img = cv2.merge((equ_b, equ_g, equ_r))#合并均衡化后的通道# 显示均衡化前后的彩色图像path2=new_path+'\\'+filecv2.imwrite(path2, equ_img)
#cv2.imshow('Original Image', img)
#cv2.imshow('Equalized Image', equ_img)
#cv2.waitKey(0)
#cv2.destroyAllWindows()

做均衡化前后的图片如下所示:

但是,由于我的图片中有些图片存在大面积白色,因此均衡之后颜色就变成了这样:

很明显,由于白色区域的影响,这个结果明显是不对的,想了各种办法:

1、比如用PS把白色区域删掉,保存成png透明背景的再跑上面的代码,结果不对

2、用PS把上面的白色区域先变成灰色,再跑,结果也还是不对

总结原因:是因为我们上面的代码是做全局直方图均衡化,并不是局部的,因此即使是透明背景,结果做出来也不对

而且为了只让我选定的区域做直方图均衡化,其他区域不变

就需要用到下面的方法

基于掩模的特定区域直方图均衡化

这里也是参考了这位博主的文章

但这里他是对灰度图像做的,我们要做的是RGB影像

因此我对代码进行了修改,可以对RGB进行基于掩模的特定区域颜色直方图均衡化

话不多说

直接看代码

import cv2
import os
import numpy as npfilePath = r"F:\deepl\sample\complete\road3\white"  # 用于获取文件名称列表
new_path = r"F:\deepl\sample\complete\road3\white-rgb"  # 目标文件夹
mask_path = r"F:\deepl\sample\complete\road3\white-mask"  # 目标文件夹
#move_path = r"F:\deepl\sample\complete\water2\sat"  # 目标文件夹file_names = filter(lambda x: x.find('png')!=-1, os.listdir(filePath))# print(file_list)for file in file_names:
# 读取彩色图像path=filePath+'\\'+filemask_path=mask_path+'\\'+fileimg = cv2.imread(path)# 分离BGR图像的通道b, g, r = cv2.split(img)mask = cv2.imread(mask_path, 0)coord = np.where(mask == 255)print(coord)b_mask = b[coord]g_mask = g[coord]r_mask = r[coord]# 对每个通道进行直方图均衡化equ_b = cv2.equalizeHist(b_mask)equ_g = cv2.equalizeHist(g_mask)equ_r = cv2.equalizeHist(r_mask)equ_img = cv2.merge((equ_b, equ_g, equ_r))#合并均衡化后的通道img2 = img.copy()for i, C in enumerate(zip(coord[0], coord[1])):img2[C[0], C[1]] = equ_img[i][0]# 显示均衡化前后的彩色图像path2=new_path+'\\'+filecv2.imwrite(path2, img2)
#cv2.imshow('Original Image', img)
#cv2.imshow('Equalized Image', equ_img)
#cv2.waitKey(0)
#cv2.destroyAllWindows()

上面3张图分别是RGB原图,黑白掩模,均衡化后的结果

这下可以看出,我们只对其中一部分指定区域做了均衡化

成功!

但是如何批量化跑呢?

我想要让程序自动从文件夹中读取图片,自动将白色和非白色区域生成掩模,然后自动读取后制作均衡化后的结果

这篇关于深度学习中RGB影像图的直方图均衡化python代码and对图片中指定部分做基于掩模的特定区域直方图均衡化的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/641758

相关文章

Java学习手册之Filter和Listener使用方法

《Java学习手册之Filter和Listener使用方法》:本文主要介绍Java学习手册之Filter和Listener使用方法的相关资料,Filter是一种拦截器,可以在请求到达Servl... 目录一、Filter(过滤器)1. Filter 的工作原理2. Filter 的配置与使用二、Listen

如何使用 Python 读取 Excel 数据

《如何使用Python读取Excel数据》:本文主要介绍使用Python读取Excel数据的详细教程,通过pandas和openpyxl,你可以轻松读取Excel文件,并进行各种数据处理操... 目录使用 python 读取 Excel 数据的详细教程1. 安装必要的依赖2. 读取 Excel 文件3. 读

Python的time模块一些常用功能(各种与时间相关的函数)

《Python的time模块一些常用功能(各种与时间相关的函数)》Python的time模块提供了各种与时间相关的函数,包括获取当前时间、处理时间间隔、执行时间测量等,:本文主要介绍Python的... 目录1. 获取当前时间2. 时间格式化3. 延时执行4. 时间戳运算5. 计算代码执行时间6. 转换为指

利用Python调试串口的示例代码

《利用Python调试串口的示例代码》在嵌入式开发、物联网设备调试过程中,串口通信是最基础的调试手段本文将带你用Python+ttkbootstrap打造一款高颜值、多功能的串口调试助手,需要的可以了... 目录概述:为什么需要专业的串口调试工具项目架构设计1.1 技术栈选型1.2 关键类说明1.3 线程模

关于MongoDB图片URL存储异常问题以及解决

《关于MongoDB图片URL存储异常问题以及解决》:本文主要介绍关于MongoDB图片URL存储异常问题以及解决方案,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐... 目录MongoDB图片URL存储异常问题项目场景问题描述原因分析解决方案预防措施js总结MongoDB图

Python ZIP文件操作技巧详解

《PythonZIP文件操作技巧详解》在数据处理和系统开发中,ZIP文件操作是开发者必须掌握的核心技能,Python标准库提供的zipfile模块以简洁的API和跨平台特性,成为处理ZIP文件的首选... 目录一、ZIP文件操作基础三板斧1.1 创建压缩包1.2 解压操作1.3 文件遍历与信息获取二、进阶技

Python Transformers库(NLP处理库)案例代码讲解

《PythonTransformers库(NLP处理库)案例代码讲解》本文介绍transformers库的全面讲解,包含基础知识、高级用法、案例代码及学习路径,内容经过组织,适合不同阶段的学习者,对... 目录一、基础知识1. Transformers 库简介2. 安装与环境配置3. 快速上手示例二、核心模

Python正则表达式语法及re模块中的常用函数详解

《Python正则表达式语法及re模块中的常用函数详解》这篇文章主要给大家介绍了关于Python正则表达式语法及re模块中常用函数的相关资料,正则表达式是一种强大的字符串处理工具,可以用于匹配、切分、... 目录概念、作用和步骤语法re模块中的常用函数总结 概念、作用和步骤概念: 本身也是一个字符串,其中

Python使用getopt处理命令行参数示例解析(最佳实践)

《Python使用getopt处理命令行参数示例解析(最佳实践)》getopt模块是Python标准库中一个简单但强大的命令行参数处理工具,它特别适合那些需要快速实现基本命令行参数解析的场景,或者需要... 目录为什么需要处理命令行参数?getopt模块基础实际应用示例与其他参数处理方式的比较常见问http

python实现svg图片转换为png和gif

《python实现svg图片转换为png和gif》这篇文章主要为大家详细介绍了python如何实现将svg图片格式转换为png和gif,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录python实现svg图片转换为png和gifpython实现图片格式之间的相互转换延展:基于Py