Matting tool是一款功能十分強(qiáng)大的無綠幕摳圖軟件，該軟件由論壇大神開發(fā)制作，采用了智能AI引擎打造，能夠幫助用戶對圖片、視頻等媒體文件進(jìn)行AI智能處理，操作簡單易上手，有需要的小伙伴千萬不要錯過哦。

【使用說明】

　　直接打開根目錄下的Matting tool by CS.exe。

　　1.Data選擇

　　Input Dir：選擇所需處理的視頻、圖片；

　　Output Dir：選擇保存位置文件夾；

　　Model weight：模型權(quán)重，選擇weights文件夾下的model.onnx；

　　2.Advance設(shè)置

　　Show Result：可以實(shí)時顯示處理過程（但由于顯示控件問題，彈出顯示了），但是由于cv2顯示會影響處理速度，可以考慮去掉勾勾；

　　mode：選擇處理的是視頻還是圖片；

　　background color：選擇摳圖背景顏色，默認(rèn)綠幕；

　　3.Start Run

　　點(diǎn)擊Start Run按鈕開始處理，可通過命令行窗口查看當(dāng)前處理進(jìn)度（注意處理時不要關(guān)閉這個窗口）。如果不想處理了，可以點(diǎn)擊Close Run按鈕結(jié)束處理進(jìn)程。

　　最終輸出為output(+audio).mp4視頻或者out.jpg圖片。

【注意事項(xiàng)】

　　原代碼為pytorch環(huán)境上訓(xùn)練的類Unet結(jié)構(gòu)模型。主要分類編碼、解碼與上采樣輸出幾個部分，具體不再贅述。

　　訓(xùn)練所使用的數(shù)據(jù)集包括：人像語義分割數(shù)據(jù)集、matting摳圖數(shù)據(jù)集。采用的損失包括了：alpha通道L1損失與拉普拉斯金字塔損失、時序相關(guān)性損失、fg部分的L1損失和時序相關(guān)損失。將模型權(quán)重文件轉(zhuǎn)換為onnx模型，本來想做個量化，所以現(xiàn)在處理時間可能稍久···，所以可以考慮后續(xù)加快預(yù)測處理速度，增加半精度選擇、模型剪枝量化處理吧。