【开源】基于RMBG的一键抠图与证件照制作系统【含一键启动包】

《博主简介》

小伙伴们好，我是阿旭。专注于人工智能、AIGC、python、计算机视觉相关分享研究。
✌更多学习资源，可关注公-仲-hao:【阿旭算法与机器学习】，共同学习交流~
👍感谢小伙伴们点赞、关注！

《------往期经典推荐------》

一、AI应用软件开发实战专栏【链接】

项目名称项目名称
1.【人脸识别与管理系统开发】2.【车牌识别与自动收费管理系统开发】
3.【手势识别系统开发】4.【人脸面部活体检测系统开发】
5.【图片风格快速迁移软件开发】6.【人脸表表情识别系统】
7.【YOLOv8多目标识别与自动标注软件开发】8.【基于YOLOv8深度学习的行人跌倒检测系统】
9.【基于YOLOv8深度学习的PCB板缺陷检测系统】10.【基于YOLOv8深度学习的生活垃圾分类目标检测系统】
11.【基于YOLOv8深度学习的安全帽目标检测系统】12.【基于YOLOv8深度学习的120种犬类检测与识别系统】
13.【基于YOLOv8深度学习的路面坑洞检测系统】14.【基于YOLOv8深度学习的火焰烟雾检测系统】
15.【基于YOLOv8深度学习的钢材表面缺陷检测系统】16.【基于YOLOv8深度学习的舰船目标分类检测系统】
17.【基于YOLOv8深度学习的西红柿成熟度检测系统】18.【基于YOLOv8深度学习的血细胞检测与计数系统】
19.【基于YOLOv8深度学习的吸烟/抽烟行为检测系统】20.【基于YOLOv8深度学习的水稻害虫检测与识别系统】
21.【基于YOLOv8深度学习的高精度车辆行人检测与计数系统】22.【基于YOLOv8深度学习的路面标志线检测与识别系统】
23.【基于YOLOv8深度学习的智能小麦害虫检测识别系统】24.【基于YOLOv8深度学习的智能玉米害虫检测识别系统】
25.【基于YOLOv8深度学习的200种鸟类智能检测与识别系统】26.【基于YOLOv8深度学习的45种交通标志智能检测与识别系统】
27.【基于YOLOv8深度学习的人脸面部表情识别系统】28.【基于YOLOv8深度学习的苹果叶片病害智能诊断系统】
29.【基于YOLOv8深度学习的智能肺炎诊断系统】30.【基于YOLOv8深度学习的葡萄簇目标检测系统】
31.【基于YOLOv8深度学习的100种中草药智能识别系统】32.【基于YOLOv8深度学习的102种花卉智能识别系统】
33.【基于YOLOv8深度学习的100种蝴蝶智能识别系统】34.【基于YOLOv8深度学习的水稻叶片病害智能诊断系统】
35.【基于YOLOv8与ByteTrack的车辆行人多目标检测与追踪系统】36.【基于YOLOv8深度学习的智能草莓病害检测与分割系统】
37.【基于YOLOv8深度学习的复杂场景下船舶目标检测系统】38.【基于YOLOv8深度学习的农作物幼苗与杂草检测系统】
39.【基于YOLOv8深度学习的智能道路裂缝检测与分析系统】40.【基于YOLOv8深度学习的葡萄病害智能诊断与防治系统】
41.【基于YOLOv8深度学习的遥感地理空间物体检测系统】42.【基于YOLOv8深度学习的无人机视角地面物体检测系统】
43.【基于YOLOv8深度学习的木薯病害智能诊断与防治系统】44.【基于YOLOv8深度学习的野外火焰烟雾检测系统】
45.【基于YOLOv8深度学习的脑肿瘤智能检测系统】46.【基于YOLOv8深度学习的玉米叶片病害智能诊断与防治系统】
47.【基于YOLOv8深度学习的橙子病害智能诊断与防治系统】48.【车辆检测追踪与流量计数系统】
49.【行人检测追踪与双向流量计数系统】50.【基于YOLOv8深度学习的反光衣检测与预警系统】
51.【危险区域人员闯入检测与报警系统】52.【高密度人脸智能检测与统计系统】

二、机器学习实战专栏【链接】，已更新31期，欢迎关注，持续更新中~~
三、深度学习【Pytorch】专栏【链接】
四、【Stable Diffusion绘画系列】专栏【链接】
五、YOLOv8改进专栏【链接】，持续更新中~~
六、YOLO性能对比专栏【链接】，持续更新中~

《------正文------》

背景

当前AI抠图技术已经非常成熟，并且有效果非常好的开源模型。 日常中可以用于物品抠图、替换证件照背景等场景，但是网上许多的证件照替换背景竟然还需要收费。今天就给大家分享一个自己制作的可以一键启动的【一键抠图与制作证件照的系统】，感兴趣的小伙伴可以自行下载体验，感谢大家的点赞关注。

效果如下：
在这里插入图片描述
在这里插入图片描述

项目实现思路

本项目主要是基于RMBG进行实现的，使用onnx模型进行目标抠出，然后对背景进行处理替换颜色。核心代码如下：

定义相关函数

首先我们定义一个抠图函数rmbg和替换背景颜色函数add_background_to_image如下所示：
抠图函数rmbg代码：

def rmbg(input_image_path,  background_color, out_size_w, out_size_h, size_opt):if size_opt == "保持原图大小":shape = cv2.imread(input_image_path).shapeout_size = (int(shape[0]),int(shape[1]))else:out_size = (int(out_size_w), int(out_size_h))print('input_image_path:',input_image_path)name = os.path.basename(input_image_path)path = os.path.dirname(input_image_path)filename, ext = os.path.splitext(name)ext = '.png'new_filename = filename + "_rmgb" + ext # 修改文件名out_path = path + new_filename # 抠图new_filename = filename + "_bg" + extoutput_image_path = path + new_filename # 证件照net = BriaRMBG_ONNX(f"rmbg/onnx/model.onnx")# prepare inputmodel_input_size = [1024,1024]orig_im = io.imread(input_image_path)orig_im_size = orig_im.shape[0:2]image = preprocess_image(orig_im, model_input_size)image = np.transpose(image, (0, 3, 1, 2))  # ONNX通常需要CHW格式# inferenceresult = net(image)# post processresult_image = postprocess_image(result[0][0], orig_im_size)# save resultpil_im = Image.fromarray(result_image)no_bg_image = Image.new("RGBA", pil_im.size, (0,0,0,0))orig_image = Image.open(input_image_path)no_bg_image.paste(orig_image, mask=pil_im)no_bg_image.save(out_path, "PNG")print(background_color, out_size)add_background_to_image(out_path, output_image_path, background_color, out_size)return out_path, output_image_path

替换背景颜色函数代码如下：

def add_background_to_image(input_image_path, output_image_path, background_color, out_size=None):"""给透明背景的PNG人像图像添加任意颜色的背景。:param input_image_path: 输入图像的路径:param output_image_path: 输出图像的路径:param background_color: 背景颜色 (R, G, B):param size: 输出图像的大小 (width, height) 默认与输入图像相同"""# 打开输入图像image = Image.open(input_image_path)# 如果图像不是PNG格式，先转换为PNGif image.format != 'PNG':image = image.convert('RGBA')if out_size is None:out_size = image.sizeout_image = Image.new('RGB', image.size, background_color)out_image.paste(image, (0,0), image)out_image.resize(out_size)# 保存新的图像out_image.save(output_image_path)

调用函数进行抠图

import cv2
import matplotlib.pyplot as plt# 输入图片的路径
input_img = 'test1.jpg'# 证件照的背景颜色
# color = "#FFFFFF" # 白色（用于护照、签证、身份证等）
color = "#438EDB" # 蓝色（用于毕业证、工作证等）
# color = "#FF0000" # 红色（用于一些特殊的证件照）# 证件照的大小
width = 295
height = 413  # 一寸（295像素 x 413像素）# 是否保持原图大小 
# size_opt = "不保持原图大小"
size_opt = "保持原图大小" # 如果选了这个会保持输入图片的大小，忽略上面的 证件照的大小 参数# color, width, height 这三个参数不影响抠图，只会影响证件照的结果
out_path, output_image_path = rmbg(input_img, color, width, height, size_opt)print('抠图后的图片： ', out_path)
print('证件照： ', output_image_path)

原始图片：
在这里插入图片描述
抠图后的图片：
在这里插入图片描述
替换背景颜色后的图片：
在这里插入图片描述

一键启动使用说明

下载项目文件后，直接双击运行一键启动.exe文件。
在这里插入图片描述

之后会在浏览器自动打开，如下网页：
在这里插入图片描述
使用步骤：
1.在背景颜色区域自行选择需要使用的背景颜色;
2.在证件尺寸大小区域选择需要的证件尺寸大小；
3.上传需要进行抠图的图片，然后点击提交按钮。
即可获取所抠出的图像，与替换背景后的图像，结果如下所示：
在这里插入图片描述
点击每张图片结果的右上角的下载按钮，即可下载生成后的图片。
在这里插入图片描述

该项目不仅可以轻松抠出人物图像，也可以实现其他物品的抠图，效果如下，小伙伴们可以自行体验。
在这里插入图片描述

关注文末名片G-Z-H：【阿旭算法与机器学习】，发送【开源】即可免费获取

在这里插入图片描述

好了，这篇文章就介绍到这里，喜欢的小伙伴感谢给点个赞和关注，更多精彩内容持续更新~~
关于本篇文章大家有任何建议或意见，欢迎在评论区留言交流！

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://xiahunao.cn/news/3223595.html

如若内容造成侵权/违法违规/事实不符，请联系瞎胡闹网进行投诉反馈，一经查实，立即删除！

相关文章

Transformer-LSTM预测 | Matlab实现Transformer-LSTM时间序列预测

Transformer-LSTM预测 | Matlab实现Transformer-LSTM时间序列预测

Transformer-LSTM预测 | Matlab实现Transformer-LSTM时间序列预测 目录 Transformer-LSTM预测 | Matlab实现Transformer-LSTM时间序列预测效果一览基本介绍程序设计参考资料 效果一览 基本介绍 1.Matlab实现Transformer-LSTM时间序列预测，Transformer-LSTM&#xf…
阅读更多...
第五次作业（多表联合查询）

第五次作业（多表联合查询）

新增员工表emp和部门表dept create table dept (dept1 int ,dept_name varchar(11)) charsetutf8; create table emp (sid int ,name varchar(11),age int,worktime_start date,incoming int,dept2 int) charsetutf8; insert into dept values (101,财务), (102,销售…
阅读更多...
入门PHP就来我这（高级）18 ~ 获取结果集

入门PHP就来我这（高级）18 ~ 获取结果集

有胆量你就来跟着路老师卷起来！ -- 纯干货，技术知识分享 路老师给大家分享PHP语言的知识了，旨在想让大家入门PHP，并深入了解PHP语言。 1 PDO中获取结果集 在PDO中获取结果集常用3中方法：fetch()方法、fetchAll()方法和…
阅读更多...
两位软件工程师创业，开发出一款软件质量保证自主AI代理，融资总额超3000万美元

两位软件工程师创业，开发出一款软件质量保证自主AI代理，融资总额超3000万美元

编译整理｜TesterHome社区 来源｜TechCrunch 随着生成式人工智能（generative AI）的出现，人工智能应用正在改变和重塑各行各业，并改变人们的工作方式。软件开发也不例外。 总部位于旧金山和东京的初创公司Aut…
阅读更多...
算法学习笔记（8.2）-动态规划入门进阶

算法学习笔记（8.2）-动态规划入门进阶

目录 问题判断: 问题求解步骤： 图例： 解析： 方法一：暴力搜索 实现代码如下所示： 解析： 方法二：记忆化搜索 代码示例： 解析： 方法三：动态规划 空间…
阅读更多...
C++进阶：继承和多态

C++进阶：继承和多态

文章目录 ❤️继承🩷继承与友元🧡继承和静态成员💛菱形继承及菱形虚拟继承💚继承和组合 ❤️多态🩷什么是多态？🧡多态的定义以及实现💛虚函数💚虚函数的重写&#x1f499…
阅读更多...
MySQL之基本查询(上)-表的增删查改

MySQL之基本查询(上)-表的增删查改

目录 Create(创建) 案例建表 插入 单行数据 指定列插入 单行数据 全列插入 多行数据 全列插入 插入是否更新 插入时更新 替换 Retrieve(读取) 建表插入 select列 全列查询 指定列查询 查询字段为表达式 为查询结果指定别名 结果去重 where条件 比较运算符 逻辑运…
阅读更多...
【深度学习基础】MacOS PyCharm连接远程服务器

【深度学习基础】MacOS PyCharm连接远程服务器

目录 一、需求描述二、建立与服务器的远程连接1. 新版Pycharm的界面有什么不同？2. 创建远程连接3. 建立本地项目与远程服务器项目之间的路径映射4.设置保存自动上传文件 三、设置解释器总结 写在前面，本人用的是Macbook Pro， M3 MAX处理器&am…
阅读更多...
opencv读取视频文件夹内视频的名字_时长_帧率_分辨率写入excel-cnblog

opencv读取视频文件夹内视频的名字_时长_帧率_分辨率写入excel-cnblog

看视频的时候有的视频文件名贼长。想要翻看，在文件夹里根本显示不出来，缩短又会丢失一些信息，所以我写了一份Python代码，直接获取视频的名字，时长，帧率，还有分辨率写到excel里。 实际效果如下图…
阅读更多...
uniapp父页面调用子页面 组件方法记录

uniapp父页面调用子页面 组件方法记录

文章目录 导文如何点击父页面，触发子页面函数先写一个子页面的基础内容父元素 如何点击父页面，修改子页面的值先写一个子页面的基础内容父元素 导文 如何点击父页面，触发子页面函数？ 如何点击父页面，修改子页面的值&am…
阅读更多...
Java基础语法--基本数据类型

Java基础语法--基本数据类型

Java基础语法–基本数据类型 Java是一种静态类型语言，这意味着每个变量在使用前都必须声明其数据类型。Java提供了多种基本数据类型，用于存储整数、浮点数、字符和布尔值等。以下是Java中的基本数据类型及其特点： 1. 整型（Integ…
阅读更多...
offer题目33:判断是否是二叉搜索树的后序遍历序列

offer题目33:判断是否是二叉搜索树的后序遍历序列

题目描述：输入一个整数数组，判断该数组是不是某二叉搜索树的后序遍历结果。如果是则返回true,否则返回false。假设输入的数组的任意两个数字都互不相同。例如，输入数组{5,7,6,9,11,10,8},则返回true,，因为这个整数是下图二叉搜索树…
阅读更多...
如何从 Vue 2 无痛升级到 Vue 3，一文搞定！

如何从 Vue 2 无痛升级到 Vue 3，一文搞定！

大家好，我是CodeQi！ 一位热衷于技术分享的码仔。 随着 Vue 3 的发布，许多开发者都面临着从 Vue 2 升级到 Vue 3 的挑战。 本文将详细介绍如何从 Vue 2 无痛升级到 Vue 3，包括每个步骤的详细说明与代码示例。 让我们开始吧！ 准备工作 在正式开始升级之前，请确保你已经…
阅读更多...
C基础day7

C基础day7

一、思维导图 二、课后练习 1、提示并输入一个字符串&#xff0c;统计该字符串中字母、数字、空格以及其他字符的个数 #include<myhead.h> #define M 20 int main(int argc, const char *argv[]) {int sum_a0,sum_b0,sum_c0,sum_d0;char str[M];printf("please en…
阅读更多...
使用redis进行短信登录验证（验证码打印在控制台）

使用redis进行短信登录验证（验证码打印在控制台）

使用redis进行短信登录验证 一、流程1. 总体流程图2. 流程文字讲解&#xff1a;3.代码3.1 UserServiceImpl&#xff1a;&#xff08;难点&#xff09;3.2 拦截器LoginInterceptor&#xff1a;3.3 拦截器配置类&#xff1a; 4 功能实现&#xff0c;成功存入redis &#xff08;黑…
阅读更多...
HTML5+CSS3小实例：响应式漫画网格布局

HTML5+CSS3小实例：响应式漫画网格布局

实例：响应式漫画网格布局 技术栈：HTML+CSS 效果： 源码： 【HTML】 <!DOCTYPE html> <html lang="zh-CN"> <head><meta charset="UTF-8"><meta name="viewport" content="width=device-width, initial-sc…
阅读更多...
Windows11配置WSL2支持代理上网

Windows11配置WSL2支持代理上网

提示&#xff1a;文章写完后&#xff0c;目录可以自动生成&#xff0c;如何生成可参考右边的帮助文档 文章目录 前言一、安装WSL2分发版二、配置步骤三、测试总结 前言 说起来本来这个功能我也不需要的&#xff0c;只是最近突然有个需求就顺便研究了下&#xff0c;WSL2默认的网…
阅读更多...
Docker-compse的应用

Docker-compse的应用

1 docker-compose # 使用了docker 面临一个比较大的问题&#xff0c;如果一个djagno项目&#xff0c;使用mysql&#xff0c;redis&#xff0c;不要一次性把所有服务都放到一个容器中&#xff0c;每个服务一个容器&#xff0c;批量的管理多个容器&#xff0c;比较难以操作&…
阅读更多...
KIVY Button¶

KIVY Button¶

Button — Kivy 2.3.0 documentation Button Jump to API ⇓ Module: kivy.uix.button Added in 1.0.0 The Button is a Label with associated actions that are triggered when the button is pressed (or released after a click/touch). To configure the button, the s…
阅读更多...
vue 切换主题色切换主题色切换主题色切换主题色切换主题色

vue 切换主题色切换主题色切换主题色切换主题色切换主题色

第一种&#xff1a;使用CSS变量 CSS变量&#xff08;Custom Properties&#xff09;是CSS的一种新特性 1.实现需求&#xff1a;自定义颜色 定义变量 全局的theme.css :root {--primary-color:red; }在组件中使用这些变量 demo.vue <template><div class"main…
阅读更多...
最新文章

Copyright @ 2022~2023