> 文章列表 > 开发了一个抠图/去背景应用

开发了一个抠图/去背景应用

开发了一个抠图/去背景应用

jr们早上好

iPhone 的 iOS 16有个很酷的功能,长按照片就能把其中的拍摄主体提取出来,抠图过程比一般的抠图App方便,精细度也更高。

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-KIlpLyow-1680141413142)(https://my-wechat.oss-cn-beijing.aliyuncs.com/%E8%8B%B9%E6%9E%9C%E6%8A%A0%E5%9B%BE1.gif)]

最近发现一个Python库——rembg,后台引擎是用于显著对象检测的深度网络架构 U²-Net

开发了一个抠图/去背景应用

所以我就用它开发了一个在线抠图应用

依然是依托Gradio + huggingface 的space

大家先来感受一下,huggingface 最近貌似不太稳

https://huggingface.co/spaces/beihai/Remove-Background-By-U2Net

开发了一个抠图/去背景应用

用法非常简单:
1 上传需要去背景的图片,点击Submit
2 稍等片刻,右侧Output将去背景后的主体另存为即可。

开发了一个抠图/去背景应用

抠图应该算蛮成熟的应用了,不过学以致用嘛。

自己动手做一个大家不觉得这很酷吗?作为一名理工男我觉得这太酷了,很符合我对未来生活的想象,科技并带着趣味。/手动狗头

实现过程实在不想再讲了,因为我之前在腾讯的这个算法,我搬到了网上,随便玩! 还有 LightGBM 可视化调参这两篇文章中已经讲过两遍了:

step1:注册Huggingface账号

step2:创建Space,SDK选Gradio

step3:克隆新建的space代码,然后将改好的代码push上去

这里只展示下核心代码吧,也是简单的离谱

核心代码

import os
os.system("/usr/local/bin/python -m pip install --upgrade pip")
import gradio as gr
from rembg import remove
import cv2def inference(img):input_img = cv2.imread(img)output = remove(input_img[:, :, [2,1,0]])return outputgr.Interface(inference, gr.inputs.Image(type="filepath", label="Input"), gr.outputs.Image(type="pil", label="Output")).launch() 

给专业选手的 Python 笔记

Python办公自动化全套代码

南京大学的Python数据分析课件

经典教材《统计学习导论Python版

87页新书《给数据科学家的python技能秘籍pdf