《博主简介》
小伙伴们好，我是阿旭。专注于人工智能、AIGC、python、计算机视觉相关分享研究。
✌更多学习资源，可关注公-仲-hao:【阿旭算法与机器学习】，共同学习交流~
👍感谢小伙伴们点赞、关注！
《------往期经典推荐------》
二、机器学习实战专栏【链接】，已更新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).shape
out_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" + ext
output_image_path = path + new_filename # 证件照
net = BriaRMBG_ONNX(f"rmbg/onnx/model.onnx")
# prepare input
model_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格式
# inference
result = net(image)
# post process
result_image = postprocess_image(result[0][0], orig_im_size)
# save result
pil_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格式，先转换为PNG
if image.format != 'PNG':
image = image.convert('RGBA')
if out_size is None:
out_size = image.size
out_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.上传需要进行抠图的图片，然后点击提交按钮。
即可获取所抠出的图像，与替换背景后的图像，结果如下所示：
点击每张图片结果的右上角的下载按钮，即可下载生成后的图片。
该项目不仅可以轻松抠出人物图像，也可以实现其他物品的抠图，效果如下，小伙伴们可以自行体验。
好了，这篇文章就介绍到这里，喜欢的小伙伴感谢给点个赞和关注，更多精彩内容持续更新~~
关于本篇文章大家有任何建议或意见，欢迎在评论区留言交流！
