一行代码实现灰度图抠图
抠图是ps的最基本技能,利用python可以实现用一行代码实现灰度图抠图。
基础算法是确定图像二值化分割阈值的大津法,将图像分成背景和前景两部分,最大化背景和前景之间的类间方差。
具体理论部分可以搜索大津法了解一下,代码部分只要在调用threshold的函数中,参数选择THRESH_OTSU就可以调用大津法分割
ret1, th1 = cv2.threshold(gray, 0, 255, cv2.THRESH_OTSU)
全部代码如下,将原图、直方图和分割后的图放在一起比较展示
image = cv2.imread("pictures/dajin.jpg")
#先转化为灰度图
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
plt.subplot(131), plt.imshow(image, "image")
plt.title("source image"), plt.xticks([]), plt.yticks([])
plt.subplot(132), plt.hist(image.ravel(), 256)
plt.title("Histogram")
ret1, th1 = cv2.threshold(gray, 0, 255, cv2.THRESH_OTSU) #方法选择为THRESH_OTSU
plt.subplot(133), plt.imshow(th1, "gray")
plt.title("OTSU,threshold is " + str(ret1)), plt.xticks([]), plt.yticks([])
plt.show()
抠图结果如下
但是这种算法对噪音和目标大小十分敏感,背景和前景像素接近的时候分割效果并不好
Python抠图程序源码
零基础学习python, 听了若干节入门免费课,即兴写了个程序,作为练习,实现抠图。
抠图的手工操作,是在PS软件中,将图片复制新建粘贴,主体保留,周围作为背景擦除,
被擦除区域,透明格式。成功后保存成 png 格式文件备用(png格式允许存储图层,保持透明区域)。手工操作费力费时,操作不当需要返回重新擦除,并且擦除精度随操作者熟练程度各异。擦除一个人物对象的周边,熟练者也需要几分钟。
python 语言,可以方便调用第三方功能包,实现抠图目的,主体对象周边比较平滑,程序执行仅需要几秒钟,大大节约了人力和时间。
准备
抠图需要第三方功能包 removebg,安装后才可以操作。安装功能包需要在Windows 的dos 窗口,win+r 热键调出运行对话框,输入 cmd 命令,打开 dos 窗口。dos窗口输入:
pip install removebg<回车> (等待下载安装结束) os包、time 包为系统自带,无需安装。
RemoveBg函数调用,需要APIKEY 实参(可称密码),这是一个多位字符串。获取需要在 url = www.remove.bg 网站注册,获得免费APIKEY码。尴尬的是,每日仅允许处理50张图片。
程序构思
1)要求允许处理多张jpg格式图片,显示运行进度, 适当的用户交互, 转成exe发布。
2)关键代码,处理单张图片的代码仅两行:
rmbg = RemoveBg('apiKey码', '错误信息记录文件名') #56行
向网站发出获取API 调用对象的请求,返回对象
rmbg.remove_background_from_img_file('./picture/' + item) #74行
使用 rmbg 对象的方法,抠一张图。
3)检查图片存放位置(文件夹存在否, 不存在则创建文件夹并退出)
4)检查文件夹中有没有文件,没有则显示信息并退出, 有则重新建表剔除非jpg格式文件。
5)检查与处理抠图定义了多个函数,避免程序条理混乱,不易阅读
readkey()
-- 读取APIkey 码文件;founddir()
-- 检查存放图片文件的文件夹是否存在,可发出提示信息;Cutout()
-- 抠图,含检查图片文件是否存在,以及剔除非jpg格式文件,并有提示;
6)主程序段,位于最后,明确提示操作准备,交互是否继续,或取消。
7) 打包成 exe 文件(略)
完整代码截图如下,每条语句均有功能注释
1 引用有关处理包
2 定义一个函数,自附带文件读取 apikye 码值
3. 定义搜索目录的函数,返回真/假。
4 定义抠图处理的主函数
4 主程序
以上为个人经验,希望能给大家一个参考,也希望大家多多支持编程网。