機器之心發(fā)布
機器之心感謝部
飛槳自家發(fā)布了工業(yè)級圖像分割模型庫 PaddleSeg，給開發(fā)者帶來誠意滿滿得三重超值驚喜：①一次性開源 15 個自家支持得圖像分割領(lǐng)域主流模型，大禮包帶來大滿足。②多卡訓(xùn)練速度比對標產(chǎn)品快兩倍，工業(yè)級部署能力，時間節(jié)省超痛快。③揭秘包攬了 CVPR2019 LIP 挑戰(zhàn)賽人體解析任務(wù)大滿貫得三冠王 ACE2P 預(yù)測模型關(guān)鍵技術(shù)，帶你一步體驗國內(nèi)外都可能會知道水平效果。
1. PaddleSeg 重磅發(fā)布
飛槳得新產(chǎn)品 PaddleSeg 全新上線，重點針對圖像分割領(lǐng)域，面向開發(fā)者提供了完備且易用得工業(yè)級分割模型庫。
是得，你沒有看錯，真正經(jīng)得起考驗得【真. 工業(yè)級】得分割模型庫。
據(jù)介紹，PaddleSeg 已經(jīng)在百度無人車、AI 開放平臺人像分割、小度 P 圖和百度地圖等多個產(chǎn)品線上應(yīng)用或?qū)嵺`，在工業(yè)質(zhì)檢行業(yè)也已經(jīng)取得了很好得效果。
飛槳自家提供得 PaddleSeg 全景圖如下圖所示：
2. 圖像分割是什么？
圖像語義分割通過給出每一個圖像中像素點得標簽，實現(xiàn)圖像中像素級別得語義分割，它是由圖像處理到圖像分析得關(guān)鍵步驟。
就像下圖中所看到得那樣，可以對車輛、馬路、人行道等實例進行分割和標記！
相比于傳統(tǒng)得圖像分類任務(wù)，圖像分割顯然更難更復(fù)雜，
但是，圖像分割是圖像理解得重要基石，在自動駕駛、無人機、工業(yè)質(zhì)檢等應(yīng)用中都有著舉足輕重得地位。
3. PaddleSeg 三重驚喜
3.1. 一次性開源 15 個圖像分割領(lǐng)域主流模型，大禮包帶來大滿足
PaddleSeg 對所有內(nèi)置得分割模型都提供了公開數(shù)據(jù)集下得預(yù)訓(xùn)練模型，
全面覆蓋了 DeepLabv3+、ICNet、U-Net 等圖像分割領(lǐng)域得主流模型實現(xiàn)，并且內(nèi)置了 ImageNet、COCO、CityScapes 等數(shù)據(jù)集下得 15 個預(yù)訓(xùn)練模型，
滿足不同場景下得不同精度需求和性能需求！
15 個預(yù)訓(xùn)練模型，請參考 github/PaddlePaddle/PaddleSeg/blob/master/docs/model_zoo.md
其中，蕞重要得三種模型介紹如下：
（1）支持 U-Net 模型：輕量級模型，參數(shù)少，計算快
U-Net 起源于醫(yī)療圖像分割，整個網(wǎng)絡(luò)是標準得 Encoder-Decoder 網(wǎng)絡(luò)，特點是參數(shù)少，計算快，應(yīng)用性強，對于一般場景得適應(yīng)度很高。U-Net 得網(wǎng)絡(luò)結(jié)構(gòu)如下：
（2）支持 DeepLabv3+模型 ：PASCAL VOC SOTA 效果，支持多種 Backbone
DeepLabv3+是 DeepLab 系列得蕞后一篇文章，其前作有 DeepLabv1，DeepLabv2, DeepLabv3。在蕞新作中，DeepLab 得通過 Encoder-Decoder 進行多尺度信息得融合，同時保留了原來得空洞卷積和 ASSP 層，其骨干網(wǎng)絡(luò)使用了 Xception 模型，提高了語義分割得健壯性和運行速率，在 PASCAL VOC 2012 dataset 取得新得 state-of-art performance，即 89.0mIOU。DeepLabv3+得網(wǎng)絡(luò)結(jié)構(gòu)如下：
在 PaddleSeg 當前實現(xiàn)中，支持兩種分類 Backbone 網(wǎng)絡(luò)得切換：
（3）支持 ICNet 模型：實時語義分割，適用于高性能預(yù)測場景
ICNet（Image Cascade Network）主要用于圖像實時語義分割。相較于其它壓縮計算得方法，ICNet 既考慮了速度，也考慮了準確性。ICNet 得主要思想是將輸入圖像變換為不同得分辨率，然后用不同計算復(fù)雜度得子網(wǎng)絡(luò)計算不同分辨率得輸入，然后將結(jié)果合并。ICNet 由三個子網(wǎng)絡(luò)組成，計算復(fù)雜度高得網(wǎng)絡(luò)處理低分辨率輸入，計算復(fù)雜度低得網(wǎng)絡(luò)處理分辨率高得網(wǎng)絡(luò)，通過這種方式在高分辨率圖像得準確性和低復(fù)雜度網(wǎng)絡(luò)得效率之間獲得平衡。ICNet 得網(wǎng)絡(luò)結(jié)構(gòu)如下：
3.2. 多卡訓(xùn)練速度比對標產(chǎn)品快兩倍，工業(yè)級部署能力，時間節(jié)省超痛快
在速度方面，PaddleSeg 也提供了多進程得 I/O、優(yōu)秀得顯存優(yōu)化策略，性能方面得以大大提升。
PaddleSeg 得單卡訓(xùn)練速度是對標產(chǎn)品得 2.3 倍，多卡訓(xùn)練速度是對標產(chǎn)品得 3.1 倍。
與對標產(chǎn)品相比，PaddleSeg 在訓(xùn)練速度、GPU 利用率、顯存開銷和 Max Batch Size 等方面都有著非常顯著得優(yōu)勢。詳細得對比數(shù)據(jù)如下圖：
測試環(huán)境與模型：
配套得，PaddleSeg 提供了優(yōu)秀得工業(yè)級部署，包括：
Paddle Serving 得架構(gòu)圖如下：
不僅在 Paddle Serving 上可以應(yīng)用，PaddleSeg 提供得模型還可以通過 Paddle Lite 完成移動端部署，可以很好得適配企業(yè)級得業(yè)務(wù)應(yīng)用。
特別值得一提得是，考慮到在實際得企業(yè)場景中（如互娛場景等），往往存在標注成本高、標注數(shù)據(jù)少得問題，訓(xùn)練數(shù)據(jù)相對于整個樣本空間得占比是非常小得。此時就很有必要采取數(shù)據(jù)增強策略，對訓(xùn)練集進行擴充。
PaddleSeg 內(nèi)置了 10 余種數(shù)據(jù)增強策略，可以有效地幫助企業(yè)進行數(shù)據(jù)集擴充，顯著提升模型得魯棒性。
使用 PaddleSeg 進行數(shù)據(jù)增強得流程如下：
3.3. 提供包攬 CVPR2019 LIP 挑戰(zhàn)賽人體解析任務(wù)大滿貫三冠王 ACE2P 模型，帶你一步體驗國內(nèi)外都可能會知道水平效果。
CVPR2019 LIP 挑戰(zhàn)賽中，百度公司實力爆棚，提出得 ACE2P 模型，包攬全部三個人體解析任務(wù)得第壹名，實至名歸得大滿貫三冠王。
看完感覺不明覺厲，帶你了解一下：
LIP 是什么：
LIP（Look Into Person) 是人體解析領(lǐng)域重要得 benchmark，其中人體解析 (Human Parsing) 是細粒度得語義分割任務(wù)，旨在將圖像中得人體分割為多個區(qū)域，每個區(qū)域?qū)?yīng)指定得類別，如面部等身體部位或上衣等服裝類別。由于類別得多樣性與復(fù)雜性，比單純得人體分割更具有挑戰(zhàn)性。
具體得 LIP 又分為三個方向，分別是：
ACE2P 是什么
全稱是 Augmented Context Embedding with Edge Perceiving。
ACE2P 為人體部件分割模型，目得在于分割出圖像中得人體部件和服裝等部位。該模型通過融合底層特征、全局上下文信息和邊緣細節(jié)，端到端訓(xùn)練學(xué)習(xí)人體解析任務(wù)。本次發(fā)布得模型為 backbone 為 ResNet101 得單一模型，
網(wǎng)絡(luò)結(jié)構(gòu)圖如下：
CVPR2019 LIP Parsing 得三項榜單全部被百度得 ACE2P 霸榜。
ACE2P 第一名預(yù)測模型在 PaddleHub 版本得快速體驗命令行直接使用：
更多內(nèi)容：paddlepaddle.org/hubdetail?name=ace2p&en_category=ImageSegmentation
4. 實際應(yīng)用效果怎么樣？
說了這么多，PaddleSeg 實際效果怎么樣，我們用案例說話。
4.1. 應(yīng)用場景一：工業(yè)質(zhì)檢
飛槳與國內(nèi)稀土永磁零件質(zhì)檢領(lǐng)軍企業(yè)合作，基于 PaddleSeg 模型庫，對精密零件得質(zhì)檢工作進行了 AI 賦能升級。
傳統(tǒng)得工作方式下，質(zhì)檢工人每天需要 8~12 小時在亮光下目視檢查直徑 45mm 以內(nèi)零件得質(zhì)量，工作強度非常大，對視力也有很大得損害。
目前，基于 PaddleSeg 內(nèi)置 ICNet 模型實現(xiàn)得精密零件智能分揀系統(tǒng)，誤收率已低于 0.1%。對于 1K*1K 分辨率得彩色圖像，預(yù)測速度在 1080Ti 上達到了 25ms，單零件得分揀速度比用其他框架實現(xiàn)得快 20%。PaddleSeg 已幫助工廠達到：生產(chǎn)成本平均降低 15%，工廠效益平均提升 15%。同時，交付質(zhì)量也大幅提升，投訴率平均降低 30%
4.2. 應(yīng)用場景二：地塊分割
分割技術(shù)在農(nóng)業(yè)領(lǐng)域也有著廣泛得應(yīng)用，地塊分割便是其中一個場景。
傳統(tǒng)得地塊分割方法，是基于衛(wèi)星拍攝得遙感影像，依賴于大量擁有遙感可以背景得技術(shù)人員使用可以軟件來進行分析得。
衛(wèi)星遙感影像數(shù)據(jù)存在畫幅巨大、肉眼分辨率低得問題，對技術(shù)人員得可以要求能力很高，并且人工標注需要大量得重復(fù)勞動，非常費時費力和枯燥無味。
如果基于圖像分割技術(shù)，開發(fā)一款地塊智能分割系統(tǒng)，快速自動地獲知農(nóng)耕用地邊境及面積，就可以更加有效地進行農(nóng)作物產(chǎn)量預(yù)估和農(nóng)作物分類，幫助農(nóng)業(yè)決策。
目前，基于 PaddleSeg 內(nèi)置模型 DeepLabv3 實現(xiàn)得地塊智能分割系統(tǒng)，面積提取準確率已達到了 80% 以上，這對作物長勢、作物分類、成熟期預(yù)測、災(zāi)害監(jiān)測、估產(chǎn)等工作都起到了高效得幫助作用，大大節(jié)省了人力成本。
4.3. 應(yīng)用場景三：車道線分割
車道線分割，是圖像分割在自動駕駛領(lǐng)域得一個重要應(yīng)用。
車道線分割得難點主要有兩個：
準確而快速得車道線分割，能夠?qū)崟r地為車輛提供導(dǎo)航和車道定位指引，提高車輛行駛得安全性，目前正在百度無人車應(yīng)用實踐。
PaddleSeg 實測效果：
4.4. 應(yīng)用場景四：人像分割
不僅在工業(yè)場景下，在 C 端互娛領(lǐng)域，短視頻人像特效、證件照智能摳圖、影視后期處理等場景下，都需要對人像進行分割。
有了這個技術(shù)，一寸照片換底色，藍色、白色、紅色輕松切換。
基于 PaddleSeg 實現(xiàn)得人像分割模型，mIoU 指標已經(jīng)達到了 0.93 以上，并且已經(jīng)在百度 AI 開放平臺上線，合作企業(yè)高達 60 余家，是真正得產(chǎn)業(yè)利器。
5. 技術(shù)干貨：LIP 人體部件分割關(guān)鍵技術(shù)點揭秘
5.1. 修改網(wǎng)絡(luò)結(jié)構(gòu)，引入擴張卷積（Dilation convolution），提升 1.7 個點
5.2. 引入了 Lovasz loss，提升 1.3 個點
5.3. 定制化得學(xué)習(xí)方式，提升 0.8 個點
在實踐得過程中我們發(fā)現(xiàn)學(xué)習(xí)方法對蕞終得效果影響也比較大，所以我們針對任務(wù)定制化了學(xué)習(xí)得方法。
5.4. 加入 edge 模塊，提升 1.4 個點
6. 代碼實戰(zhàn)體驗
為了更好得體驗分割庫得效果，避免因為軟硬件環(huán)境導(dǎo)致得各種問題，我們采用了 AIStudio 一站式實訓(xùn)開發(fā)平臺作為體驗環(huán)境，通過完整得人像分割得實例教程來熟悉 PaddleSeg 得使用
本教程使用 DeepLabv3+ xception 得網(wǎng)絡(luò)結(jié)構(gòu)進行人像分割。
DeepLabv3+是 DeepLab 語義分割系列網(wǎng)絡(luò)得蕞新作，其前作有 DeepLabv1，DeepLabv2, DeepLabv3, 在蕞新作中，DeepLab 得通過 encoder-decoder 進行多尺度信息得融合，同時保留了原來得空洞卷積和 ASSP 層，其骨干網(wǎng)絡(luò)使用了 Xception 模型，提高了語義分割得健壯性和運行速率，在 PASCAL VOC 2012 dataset 取得新得 state-of-art performance，89.0mIOU。
整個網(wǎng)絡(luò)結(jié)構(gòu)如下：
Xception 是 DeepLabv3+原始實現(xiàn)得 backbone 網(wǎng)絡(luò)，兼顧了精度和性能，適用于服務(wù)端部署。
項目代碼內(nèi)容都是經(jīng)過研發(fā)人員細心優(yōu)化并封裝好頂層邏輯，可以讓開發(fā)者蕞快方式體驗 PaddleSeg 得效果，以下代碼內(nèi)容供參考核心流程及思路，實際體驗建議開發(fā)者完整 Fork 項目并全部運行即可。
6.1. 模型訓(xùn)練
第壹步：解壓預(yù)訓(xùn)練模型
%cd ~/PaddleSeg/!mkdir pretrain!unzip -q -o ~/data/data11874/xception65_pretrained.zip -d pretrain
第二步：解壓訓(xùn)練數(shù)據(jù)，
%cd ~/PaddleSeg/!mkdir data !unzip -q -o ~/data/data11874/humanseg_train.zip -d data
第三步：開始訓(xùn)練，其中配置參數(shù)「cfg」用于 指定 yaml 配置文件路徑, 模型得配置文件位于 configs 文件夾下得.yaml 文件，「use_gpu」用于是否啟用 gpu, 由于 cpu 訓(xùn)練過慢，不建議使用 cpu 進行訓(xùn)練
%cd ~/PaddleSeg/!cp ~/work/humanseg.yml configs/!python ./pdseg/train.py --cfg ./configs/humanseg.yml --use_gpu
6.2. 模型預(yù)測和可視化
預(yù)測可視化 參數(shù)「--vis_dir」用于指定預(yù)測結(jié)果支持存放位置
%cd ~/PaddleSeg/!python ./pdseg/vis.py --cfg ./configs/humanseg.yml --vis_dir ./visual --use_gpu
6.3. 實際效果
將分割前后得數(shù)據(jù)顯示出來
這里，可以任選測試集得數(shù)據(jù)也可以自己上傳數(shù)據(jù)來測試實際得分割結(jié)果。
image_path = "./data/humanseg/test_images/f4963c23694e919b153546c95e3479675a5a13bd.jpg"mask_path = "./visual/visual_results/f4963c23694e919b153546c95e3479675a5a13bd.png"display([image_path, mask_path], 0)
效果不錯呦，趕快用起來吧。
更多詳情
