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

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

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

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

主要包括：

 

MobileNet v1

 

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

 

ResNet 系列模型

 

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

 

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

 

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預訓練模型使用說明書】

 

 

1. 安裝PaddlePaddle和模型庫

 

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

 

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

 

2. 加載預訓練模型

 

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

 

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

下拉到readme頁面的最下方：

 

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

 

在train.py文件中，

 

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

 

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

 

例如，加載MobileNet v1預訓練模型進行微調：

 

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

 

結語

 

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

 

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

>> 訪問 PaddlePaddle 官網，瞭解更多相關內容。