基于OpenCV python3实现证件照换背景的方法
更新时间：2019年03月22日 09:52:48 作者：ahaofan
这篇文章主要介绍了基于OpenCV python3实现证件照换背景的方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
简述
生活中经常要用到各种要求的证件照电子版，红底，蓝底，白底等，大部分情况我们只有其中一种，所以通过技术手段进行合成，用ps处理证件照，由于技术不到位，有瑕疵，所以想用python&openCV通过代码的方式实现背景颜色替换，加强一下对于openCV的学习，锻炼一下编码水平。
软件环境：
python3.5
opencv2
windows 10
图像载入
导入opencv库，使用imread函数读取图片
import cv2
import numpy as np
img=cv2.imread('zjz.jpg')
由于证件照太大，不方便显示，故进行缩放
#缩放
rows,cols,channels = img.shape
img=cv2.resize(img,None,fx=0.5,fy=0.5)
rows,cols,channels = img.shape
cv2.imshow('img',img)
原图如下
（图片源于网络，已经马赛克处理，如有侵权，私信立即删除）
获取背景区域
首先将读取的图像默认BGR格式转换为HSV格式，然后通过inRange函数获取背景的mask。
HSV颜色范围参数可调节根据这篇文章
hsv=cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
lower_blue=np.array([78,43,46])
upper_blue=np.array([110,255,255])
mask = cv2.inRange(hsv, lower_blue, upper_blue)
cv2.imshow('Mask', mask)
获得的mask如下图
如图所示蓝色的背景在图中用白色表示，白色区域就是要替换的部分，但是黑色区域内有白点干扰，所以进一步优化。
腐蚀和膨胀
#腐蚀膨胀
erode=cv2.erode(mask,None,iterations=1)
cv2.imshow('erode',erode)
dilate=cv2.dilate(erode,None,iterations=1)
cv2.imshow('dilate',dilate)
经过腐蚀和膨胀操作后如下图
处理后图像单独白色点消失。
替换背景色
遍历全部像素点，如果该颜色为dilate里面为白色（255）则说明该点所在背景区域，于是在原图img中进行颜色替换。
#遍历替换
for i in range(rows):
for j in range(cols):
if dilate[i,j]==255:
img[i,j]=(0,0,255)#此处替换颜色，为BGR通道
cv2.imshow('res',img)
最终结果如下
（图片源于网络，已经马赛克处理，如有侵权，私信立即删除）
总结
最开始想直接通过遍历全图进行替换背景色，但是图像中难免有些像素点和背景色一样，造成了干扰，导致最后结果不尽人意，所以想通过这种方法进行处理。显然最后有明显的ps痕迹。
最后贴上完整代码，不足之处欢迎各位指正！
import cv2
import numpy as np
img=cv2.imread('zjz.jpg')
#缩放
rows,cols,channels = img.shape
img=cv2.resize(img,None,fx=0.5,fy=0.5)
rows,cols,channels = img.shape
cv2.imshow('img',img)
#转换hsv
hsv=cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
lower_blue=np.array([78,43,46])
upper_blue=np.array([110,255,255])
mask = cv2.inRange(hsv, lower_blue, upper_blue)
cv2.imshow('Mask', mask)
#腐蚀膨胀
erode=cv2.erode(mask,None,iterations=1)
cv2.imshow('erode',erode)
dilate=cv2.dilate(erode,None,iterations=1)
cv2.imshow('dilate',dilate)
#遍历替换
for i in range(rows):
for j in range(cols):
if dilate[i,j]==255:
img[i,j]=(0,0,255)#此处替换颜色，为BGR通道
cv2.imshow('res',img)
cv2.waitKey(0)
cv2.destroyAllWindows()
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。
相关文章
大家好，本篇文章主要讲的是python图片格式转换脚本，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下，方便下次浏览2022-01-01
这篇文章主要介绍了django 信号调度机制详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-07-07
网站、小程序、APP 是否已经登录所代表的状态,代表一个概念是登录态, 我们常用的登录态验证方式有cookie,session,token,token提供了另外一种不需要缓存账户和密码的登录状态验证方式,本文给大家介绍了用Flask实现token登录校验的解决方案,需要的朋友可以参考下2024-03-03
这篇文章主要为大家详细介绍了python pygame实现滚动横版射击游戏城市之战，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-11-11
这篇文章主要为大家介绍了python绘制y关于x的线性回归线性方程图像实例,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-10-10
这篇文章主要介绍了python如何查找图片按钮的坐标位置，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-11-11
Pycharm主题切换(禁用)导致UI界面显示异常的解决方案
这篇文章主要介绍了Pycharm主题切换(禁用)导致UI界面显示异常的原因分析和解决方案,文中通过图文结合的方式给大家介绍的非常详细,需要的朋友可以参考下2024-06-06
在本文里我们给学习python的朋友们整理了关于python把1变成01的步骤总结内容，有需要的朋友们学习下。2019-02-02
本文主要介绍了pandas的apply函数用法详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-01-01
在Python中最重要的数据类型包括字符串、列表、元组和字典等。本篇文章主要讲述Python的字符串基础知识。下面跟着小编一起来看下吧2017-03-03
最新评论