圖像分類任務不用冷啓動,PaddlePaddle一口氣發佈十大預訓練模型

PaddlePaddle在不斷增長官方支持的模型的同時,也在關注預訓練模型的豐富度。算法

在過去的版本中,咱們已經發布了目標檢測Faster-RCNN、MobileNet-SSD、PyramidBox和場景文字識別CRNN-CTC、OCR Attention共計5個預訓練模型。網絡

近期,在圖像分類領域咱們一口氣發佈了四個系列共十個預訓練模型,豐富擴充了預訓練模型庫,助力用戶提升構建模型的效率,大大減輕「煉丹」的煩惱。架構

主要包括:框架

MobileNet v1學習

針對亟需在移動端應用深度學習技術的需求,MobileNet v1在存儲空間和能耗低的地方表現優秀,在損失精度很小的狀況下,計算量,存儲空間,準確率方面都有明顯提高。測試

ResNet 系列模型ui

ResNet創新性的提出了殘差結構,一舉在ILSVRC2015比賽中取得冠軍,top5錯誤率爲3.57%。斯坦福大學的Joyce Xu將ResNet稱爲「真正從新定義了咱們看待神經網絡的方式」的三大架構之一。spa

在最新發布的PaddlePaddle 預訓練模型包括有ResNet50,ResNet101和ResNet152。blog

SE_ResNet 系列模型圖片

SE 全稱 Sequeeze-and-Excitation,在ILSVRC 2017 的分類項目中取得 了第一名的成績。在 ImageNet 數據集上將 top-5 錯誤率從原先的最好成績 2.991% 下降到 2.251%。

在最新發布的PaddlePaddle 預訓練模型包括有SE_ResNeXt50_32x4d和SE_ResNeXt101_32x4d。

VGG 系列模型

VGGNet是牛津大學計算機視覺組和DeepMind公司共同研發一種深度卷積網絡,2014年在ILSVRC比賽上得到了分類項目的第二名和定位項目的第一名。

在最新發布的PaddlePaddle預訓練模型包括有VGG11,VGG13,VGG16,VGG19。

PaddlePaddle復現結果

以上預訓練模型均通過官方測試驗證,在精度上皆達到了應用要求。

您能夠在他們的基礎上進行開發,省卻本身訓練參數的過程,具體加載方式參考使用說明書。

圖像分類任務不用冷啓動,PaddlePaddle一口氣發佈十大預訓練模型
圖片描述
【PaddlePaddle預訓練模型使用說明書】

安裝PaddlePaddle和模型庫
圖片描述
在PaddlePaddle的模型庫中已經包含了最新的相關訓練代碼,通過簡單的配置與加載,便可快速部署研發,首先請安裝最新版的PaddlePaddle而且下載PaddlePaddle模型庫:

圖像分類任務不用冷啓動,PaddlePaddle一口氣發佈十大預訓練模型

當模型成功克隆後,您能夠在fluid/PaddleCV/image_classification下看到用於訓練的train.py代碼。

加載預訓練模型
使用咱們以前在ImageNet數據集上訓練的預訓練模型,能夠直接使用相應的結構和權重,將它們應用到您正在面對的問題上。

預訓練模型從哪裏下載呢?首先進入官方模型介紹

下拉到readme頁面的最下方:

能夠看到一個Released models的表格。在表格的model一列是模型的名稱,這個名稱是一個超連接,連接對應的是這個模型的預訓練權重下載地址,點擊模型名稱便可下載相應的預訓練模型。

在train.py文件中,
圖片描述
圖像分類任務不用冷啓動,PaddlePaddle一口氣發佈十大預訓練模型

經過fluid.io.load_vars加載相關預訓練參數。

運行train.py, 經過指定 pretrained_model= "下載好的預訓練模型路徑",加載相應預訓練模型進行訓練。

例如,加載MobileNet v1預訓練模型進行微調:
圖片描述
圖像分類任務不用冷啓動,PaddlePaddle一口氣發佈十大預訓練模型

您能夠更改調整預訓練模型參數,來適應具體工做方向,例如檢測,圖像分類等。

結語

對於想學習算法或者嘗試現有框架的人來講,使用預訓練模型可以提供很好的幫助。由於時間與計算量方面的諸多限制,不方便隨時從頭開始訓練一個模型,這也正是預訓練模型存在的緣由。您能夠運用預訓練做爲基準來改進現有模型,或者根據它測試本身的模型。

年後咱們還會繼續發佈幾組新的圖像分類預訓練模型(包括GoogleNet,MobileNetV2,ShuffleNet系列等),敬請期待,也歡迎你們留言告訴咱們您最期待的預訓練模型。

相關文章
相關標籤/搜索