LeeNet分類網路(ImageNet100:top1-73.44%,top5-91.40%,CPU_forward-30ms,model_size-2.7M)
阿新 • • 發佈:2019-01-25
一、前言
分類網路應用廣泛,亦可作為預訓練模型用於檢測網路,有很好的科研和商業價值。一般的分類網路基於多種資料集做了一個綜合的測評,但是因工程條件限制,很難將所有資料集下載並測試,所以博主整理了imagenet100資料集,是從imagenet1000中扣取出來的100類,每類訓練資料1000張,測試資料300張,滿足工業應用。
二、網路
網路改自ResNet,訓練出的網路模型大小為2.7MB。
三、測試條件
測試硬體:CPU-i7-4790-8core
測試軟體:vs2013,caffe-windows,以及各種依賴庫
四、測試結果
測試結果:準確率方面(top1-73.44%,top5-91.40%)
速度
模型大小(2.7MB)
五、對比結果:
已知各網路對比結果如下(取自YOLO官網),可見博主網路在工業應用中有一定的優勢:
六、後續計劃:1)下載多個數據集對比,考慮撰寫論文
2)考慮公開網路和模型,上傳至github,以供科研交流
任何問題請加唯一QQ2258205918(名稱samylee)
!!!