最近发现 BackgroundMattingV2 项目的一些使用上的小缺陷，但是他却可以做到头发丝精细的抠图效果。我将项目稍微魔改了一下，让他在可以选择单一图片的基础上，可以把抠好的图片贴在自定义的背景图上，这样就可以让照片中的人物，出现在任何背景上。是不是很有意思？

本文的github仓库地址为：https://github.com/Hy-1990/hy_bgmatting，由于模型文件过大，没放在仓库中，本文下面有模型下载地址。

我们先看一下项目的结构，如图：

其中，model文件夹放的是模型文件，模型文件的下载地址为：https://drive.google.com/drive/folders/1NmyTItr2jRac0nLoZMeixlcU1myMiYTs

下载该模型放到model文件夹下。

依赖文件如下：

本文由技术群粉丝分享，项目源码、数据、技术交流提升，均可加交流群获取，群友已超过2000人，添加时最好的备注方式为：来源+兴趣方向，方便找到志同道合的朋友

方式①、添加微信号：dkl88191，备注：来自CSDN
方式②、微信搜索公众号：Python学习与数据挖掘，后台回复：加群

我们需要准备一张照片以及照片的背景图，和你需要替换的图片。我这边选择的是BackgroundMattingV2给出的一些参考图，原始图与背景图如下：

新的背景图（我随便找的）如下：

不废话了，上核心代码。

代码说明

1、handle方法的参数一次为：原始图路径、原始背景图路径、新背景图路径。

1、我将原项目中inferance_images使用的类都移到一个文件中，精简一下项目结构。

2、ImagesDateSet我重新构造了一个新的NewImagesDateSet,，主要是因为我只打算处理一张图片。

3、最终图片都存在相同目录下，避免重复使用uuid作为文件名。

4、本文给出的代码没有对文件格式做严格校正，不是很关键，如果需要补充就行。

验证一下效果

怎么样？还是很炫吧！