1. 程式人生 > >ssd物體檢測模型訓練和測試總結

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

http max cti https 都得 硬件 obj git detection

參考網址: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相關的錯誤。
訓練模型和測試之前要做到心中有數,明確步驟,知道每一步為什麽做,
硬件要求和軟件要求都得提前考慮。

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