ssd物體檢測模型訓練和測試總結


參考網址:github:https://github.com/naisy/realtime_object_detection

2018.10.16
ssd物體檢測總結:
切記粗略地看一遍備注就開始訓練模型
出現的錯誤:1、用branch1.5,tensorflow-gpu==1.8訓練的模型在GT730,顯存2g,運行不了,tensorflow-gpu==1.5沒有NoMaxSuppressionv3,
2、用預訓練模型ssd_mobilenet_coco_2018_1_28,tensorflowgpu==1.5訓練,模型在cuda9.0,cudnn7.05上訓練,得到的模型在GT730上2fps,
3、用預訓練模型ssd_mobilenet_coco_2017_11_27,tensorflow-gpu==1.5訓練,模型在GT730上18fps,
4、同時開兩個終端,7fps,三個終端4fps,

5、選擇nms_v1還是nms_v2,在測試備注里有說明,
6、模型一部分用cudnn7.1訓練,一部分用cudnn7.05訓練會報cudnn相關的錯誤。
訓練模型和測試之前要做到心中有數,明確步驟,知道每一步為什么做,
硬件要求和軟件要求都得提前考慮。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM