亲宝软件园·资讯

展开

python灰度图抠图

我头秃了a 人气:0

一行代码实现灰度图抠图

抠图是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)检查与处理抠图定义了多个函数,避免程序条理混乱,不易阅读

6)主程序段,位于最后,明确提示操作准备,交互是否继续,或取消。

7) 打包成 exe 文件(略)

完整代码截图如下,每条语句均有功能注释

1 引用有关处理包

2 定义一个函数,自附带文件读取 apikye 码值

3. 定义搜索目录的函数,返回真/假。

4 定义抠图处理的主函数

4 主程序

以上为个人经验,希望能给大家一个参考,也希望大家多多支持。

加载全部内容

相关教程
猜你喜欢
用户评论