Matting tool是一款功能十分强大的无绿幕抠图软件,该软件采用了智能AI引擎打造,能够帮助用户对图片、视频等媒体文件进行AI智能处理,操作简单易上手,有需要的小伙伴千万不要错过哦。原代码为pytorch环境上训练的类Unet结构模型。主要分类编码、解码与上采样输出几个部分,具体不再赘述。训练所使用的数据集包括:人像语义分割数据集、matting抠图数据集。采用的损失包括了:alpha通道L1损失与拉普拉斯金字塔损失、时序相关性损失、fg部分的L1损失和时序相关损失。将模型权重文件转换为onnx模型,本来想做个量化,所以现在处理时间可能稍久···,所以可以考虑后续加快预测处理速度,增加半精度选择、模型剪枝量化处理吧。