基于PaddlePaddle的PaddleSeg實現(xiàn)人像分隔

關于人臉的前景提取,目前做的最成熟的就是remove.bg了祥楣,商用的效果確實逆天了寿桨。大家可以看看這張圖,


人臉圖

這張圖人臉提取的難點就是頭發(fā)絲是否處理的絲滑琅坡,重點要看兩側(cè)及頭頂?shù)陌l(fā)絲悉患,看看remove.bg處理后的無底效果


無底


紅底

“外國的月亮就是圓”,商用的服務確實厲害榆俺,不過價格上不太親民售躁。在問過gpt之后,它給了一個建議谴仙,可以償試PaddleSeg迂求,地址為https://github.com/PaddlePaddle/PaddleSeg/blob/release/2.8/contrib/PP-HumanSeg/README_cn.md碾盐。仔細閱讀之后晃跺,發(fā)現(xiàn)這個開源的框架真的還不錯,“授人以魚不如授人以漁”毫玖。從模型的準備掀虎,訓練,評估付枫,預測烹玉,導出,部署阐滩,使用二打。一個不少,step by step掂榔。當然继效,現(xiàn)在只是用它的模型測試了一下症杏。目前有兩個模型,一個是肖像分割 瑞信,另個一個是通用人像分割厉颤。這里抄一下肖像分割模型:

肖像分割模型


針對手機視頻通話、Web視頻會議等實時半身人像的分割場景凡简,PP-HumanSeg發(fā)布了自研的肖像分割模型逼友。該系列模型可以開箱即用,零成本直接集成到產(chǎn)品中秤涩。

PP-HumanSegV1-Lite肖像分割模型帜乞,分割效果較好,模型體積非常小筐眷,模型結(jié)構(gòu)見鏈接挖函。

PP-HumanSegV2-Lite肖像分割模型,對比V1模型浊竟,推理速度提升45.5%怨喘、mIoU提升3.03%、可視化效果更佳振定,核心在于:

更高的分割精度:使用PaddleSeg推出的超輕量級分割模型(鏈接)必怜,具體選擇MobileNetV3作為骨干網(wǎng)絡,設計多尺度特征融合模塊(Multi-Scale Feature Aggregation Module)后频。

更快的推理速度:減小模型最佳輸入尺寸梳庆,既減少了推理耗時,又增大模型感受野卑惜。

更好的通用性:使用遷移學習的思想膏执,首先在大型通用人像分割數(shù)據(jù)集上預訓練,然后在小型肖像分割數(shù)據(jù)集上微調(diào)露久。


具體調(diào)用還是python來處理更米,我只貼一下效果圖。


紅底

相比較商用的實現(xiàn)毫痕,效果確實差了一個層次征峦。下一步考慮將該功能發(fā)布成http服務,估計要好好學習一下Flask消请,WSGI了栏笆。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市臊泰,隨后出現(xiàn)的幾起案子蛉加,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 206,126評論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件针饥,死亡現(xiàn)場離奇詭異祟偷,居然都是意外死亡,警方通過查閱死者的電腦和手機打厘,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,254評論 2 382
  • 文/潘曉璐 我一進店門修肠,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人户盯,你說我怎么就攤上這事嵌施。” “怎么了莽鸭?”我有些...
    開封第一講書人閱讀 152,445評論 0 341
  • 文/不壞的土叔 我叫張陵吗伤,是天一觀的道長。 經(jīng)常有香客問我硫眨,道長足淆,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 55,185評論 1 278
  • 正文 為了忘掉前任礁阁,我火速辦了婚禮巧号,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘姥闭。我一直安慰自己丹鸿,他們只是感情好,可當我...
    茶點故事閱讀 64,178評論 5 371
  • 文/花漫 我一把揭開白布棚品。 她就那樣靜靜地躺著靠欢,像睡著了一般。 火紅的嫁衣襯著肌膚如雪铜跑。 梳的紋絲不亂的頭發(fā)上门怪,一...
    開封第一講書人閱讀 48,970評論 1 284
  • 那天,我揣著相機與錄音锅纺,去河邊找鬼掷空。 笑死,一個胖子當著我的面吹牛伞广,可吹牛的內(nèi)容都是我干的拣帽。 我是一名探鬼主播疼电,決...
    沈念sama閱讀 38,276評論 3 399
  • 文/蒼蘭香墨 我猛地睜開眼嚼锄,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了蔽豺?” 一聲冷哼從身側(cè)響起区丑,我...
    開封第一講書人閱讀 36,927評論 0 259
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后沧侥,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體可霎,經(jīng)...
    沈念sama閱讀 43,400評論 1 300
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 35,883評論 2 323
  • 正文 我和宋清朗相戀三年宴杀,在試婚紗的時候發(fā)現(xiàn)自己被綠了癣朗。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 37,997評論 1 333
  • 序言:一個原本活蹦亂跳的男人離奇死亡旺罢,死狀恐怖旷余,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情扁达,我是刑警寧澤正卧,帶...
    沈念sama閱讀 33,646評論 4 322
  • 正文 年R本政府宣布,位于F島的核電站跪解,受9級特大地震影響炉旷,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜叉讥,卻給世界環(huán)境...
    茶點故事閱讀 39,213評論 3 307
  • 文/蒙蒙 一窘行、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧图仓,春花似錦抽高、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,204評論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至帚豪,卻和暖如春碳竟,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背狸臣。 一陣腳步聲響...
    開封第一講書人閱讀 31,423評論 1 260
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留烛亦,地道東北人。 一個月前我還...
    沈念sama閱讀 45,423評論 2 352
  • 正文 我出身青樓煤禽,卻偏偏與公主長得像,于是被迫代替她去往敵國和親檬果。 傳聞我的和親對象是個殘疾皇子唐断,可洞房花燭夜當晚...
    茶點故事閱讀 42,722評論 2 345

推薦閱讀更多精彩內(nèi)容