手把手教學(xué)圖片、視頻去背景
點(diǎn)擊下方“AI算法與圖像處理”，一起進(jìn)步！
重磅干貨，第一時(shí)間送達(dá)
環(huán)境
windows 10 64-bit python 3.8.11 backgroundremover 0.1.9
前言
backgroundremover是一款開源免費(fèi)的命令行工具，專門用來消除圖片和視頻中的背景，使用python語言編寫，項(xiàng)目地址 https://github.com/nadermx/backgroundremover。
軟件安裝
可以使用pip安裝
pip install backgroundremover
當(dāng)然，下載源碼進(jìn)行安裝也是可以的
git clone https://github.com/nadermx/backgroundremover.git
cd backgroundremover
python setup.py install
目前最新版本為0.1.9，后面測試也是基于這個(gè)版本進(jìn)行。
接下來還需要安裝ffmpeg，它是用來進(jìn)行視頻處理的。從官網(wǎng)下載編譯好的二進(jìn)制文件 http://ffmpeg.org/download.html，盡量下載較新的版本，并將其路徑加入到系統(tǒng)環(huán)境變量中
最后安裝下torch，由于本機(jī)有gpu，安裝的是gpu版本的，版本要求1.7以上
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
使用方法
準(zhǔn)備一張帶背景的原圖
執(zhí)行下面命令就可以去除背景，操作非常簡單
backgroundremover -i original.jpg -o output.png
操作過程中會(huì)到google driver去下載所需要的u2net模型，但是由于國內(nèi)的網(wǎng)絡(luò)環(huán)境，無法訪問
這時(shí)候，可以通過下面的鏈接去下載
u2net.pthu2net_human_seg.pth
另外也提供個(gè)網(wǎng)盤的地址
鏈接：https://pan.baidu.com/s/1ZYzV0SYEjlBldsGXIn2IXA
提取碼：srix
下載后的模型文件，將其放到~/.u2net文件夾下，如果文件夾不存在就自己創(chuàng)建
準(zhǔn)備就緒后，重新執(zhí)行上面的去背指令，就可以得到去掉背景的圖片了
backgroundremover可以使用三種模型來達(dá)到圖片視頻去背的目的，分別是u2netp、u2net和u2net_human_seg
比如說要使用u2net_human_seg，可以使用-m參數(shù)
backgroundremover -i original.jpg -m u2net_human_seg -o output.png
如果要處理視頻的話，可以這樣
backgroundremover -i test.mp4 -tv -o output.mov
視頻背景去除后疊加到另一個(gè)視頻上
backgroundremover -i test.mp4 -tov -tv videtobeoverlayed.mp4 -o output.mov
視頻去背后生成gif動(dòng)圖
backgroundremover -i test.mp4 -tg -o output.gif
更多場景使用及參數(shù)設(shè)置，請參考官方文檔
參考資料
https://github.com/nadermx/backgroundremover https://github.com/nadermx/backgroundremover/issues/9 https://github.com/xuebinqin/U-2-Net
交流群
歡迎加入公眾號(hào)讀者群一起和同行交流，目前有美顏、三維視覺、計(jì)算攝影、檢測、分割、識(shí)別、醫(yī)學(xué)影像、GAN、算法競賽等微信群
個(gè)人微信（如果沒有備注不拉群！） 請注明：地區(qū)+學(xué)校/企業(yè)+研究方向+昵稱
下載1：何愷明頂會(huì)分享
在「AI算法與圖像處理」公眾號(hào)后臺(tái)回復(fù)：何愷明，即可下載。總共有6份PDF，涉及 ResNet、Mask RCNN等經(jīng)典工作的總結(jié)分析
下載2：終身受益的編程指南：Google編程風(fēng)格指南
在「AI算法與圖像處理」公眾號(hào)后臺(tái)回復(fù)：c++，即可下載。歷經(jīng)十年考驗(yàn)，最權(quán)威的編程規(guī)范！
下載3 CVPR2021 在「AI算法與圖像處理」公眾號(hào)后臺(tái)回復(fù)：CVPR，即可下載1467篇CVPR 2020論文 和 CVPR 2021 最新論文
