Matting tool是由论坛专家开发创作的一款非常强大的绿屏免费抠图软件。它帮助用户通过AI智能处理照片、视频等媒体文件。简单易操作。如果您需要的话请尝试一下。

【使用说明】

直接在根目录下的CS.exe中打开抠图工具。

1、数据选择

输入目录：选择要处理的视频和图像。

输出目录：选择保存的文件夹。

模型重量：模型的重量。选择权重文件夹下的model.onnx。

2.详细设置

显示结果：可以实时显示过程（但由于显示控制问题，可能会出现弹窗）。但是，由于cv2 显示会影响处理速度，因此请考虑取消选中它。

模式：选择是否处理视频或照片。

背景颜色：选择剪切图的背景颜色。默认为绿屏。

3.开始运行

单击“开始运行”按钮启动该进程后，您可以在命令行窗口中看到当前进程的进度（注意在此过程中不要关闭该窗口）。如果您不想继续处理，请单击“关闭执行”按钮结束处理过程。

最终输出是一个output(+audio).mp4视频或一个out.jpg图像。

【注意事项】

原始代码是在pytorch环境中训练的类似Unet的结构模型。主要分为编码、解码、上采样的输出部分，具体细节我们就不解释了。

用于训练的数据集包括人像语义分割数据集和抠图数据集。使用的损失包括alpha通道中的L1损失和拉普拉斯金字塔损失、定时相关损失、fg部分中的L1损失以及定时相关损失。将模型权重文件转换为onnx模型由于我本来想做一些量化，所以处理时间可能会长一点。因此，未来可以考虑加快预测过程并添加半精度选择和模型。修剪量化过程。