1. 程式人生 > >LeeNet分類網路(ImageNet100:top1-73.44%,top5-91.40%,CPU_forward-30ms,model_size-2.7M)

LeeNet分類網路(ImageNet100:top1-73.44%,top5-91.40%,CPU_forward-30ms,model_size-2.7M)

一、前言

分類網路應用廣泛,亦可作為預訓練模型用於檢測網路,有很好的科研和商業價值。一般的分類網路基於多種資料集做了一個綜合的測評,但是因工程條件限制,很難將所有資料集下載並測試,所以博主整理了imagenet100資料集,是從imagenet1000中扣取出來的100類,每類訓練資料1000張,測試資料300張,滿足工業應用。

二、網路

網路改自ResNet,訓練出的網路模型大小為2.7MB

三、測試條件

測試硬體:CPU-i7-4790-8core

測試軟體:vs2013,caffe-windows,以及各種依賴庫

四、測試結果

測試結果:準確率方面(top1-73.44%,top5-91.40%

                 速度

方面(forward_CPU_time-30ms

                 模型大小(2.7MB

五、對比結果:

已知各網路對比結果如下(取自YOLO官網),可見博主網路在工業應用中有一定的優勢:


六、後續計劃:1)下載多個數據集對比,考慮撰寫論文

                        2)考慮公開網路和模型,上傳至github,以供科研交流

任何問題請加唯一QQ2258205918(名稱samylee)  !!!