图像处理

Matting tool by CS是一款无绿幕抠图软件，该软件可以帮助用户们对视频、图片等媒体文件进行快速的ai抠图，而且对比免费的抠图工具，它的效果还是十分不错，可以帮助用户们快速抠出自己想要的效果，因此有需求的用户，快来下载体验吧！

操作说明

直接打开根目录下的Matting tool by CS.exe。

1.Data选择

Input Dir：选择所需处理的视频、图片；

Output Dir：选择保存位置文件夹；

Model weight：模型权重，选择weights文件夹下的model.onnx；

2.Advance设置

Show Result：可以实时显示处理过程（但由于显示控件问题，弹出显示了），但是由于cv2显示会影响处理速度，可以考虑去掉勾勾；

mode：选择处理的是视频还是图片；

background color：选择抠图背景颜色，默认绿幕；

3.Start Run

点击Start Run按钮开始处理，可通过命令行窗口查看当前处理进度（注意处理时不要关闭这个窗口）。如果不想处理了，可以点击Close Run按钮结束处理进程。

最终输出为output(+audio).mp4视频或者out.jpg图片。

注意事项

原代码为pytorch环境上训练的类Unet结构模型。主要分类编码、解码与上采样输出几个部分，具体不再赘述。

训练所使用的数据集包括：人像语义分割数据集、matting抠图数据集。采用的损失包括了：alpha通道L1损失与拉普拉斯金字塔损失、时序相关性损失、fg部分的L1损失和时序相关损失。将模型权重文件转换为onnx模型，本来想做个量化，所以现在处理时间可能稍久···，所以可以考虑后续加快预测处理速度，增加半精度选择、模型剪枝量化处理吧。

更新日志

1.修复已知bug

2.优化操作体验