下載ssd:

git clone?https://github.com/conner99/caffe.git

cd caffe

git checkout ssd-microsoft

修改CommonSettings.props

在caffe-master\src\caffe下新建3rdparty文件夾，里面添加hungarian.cpp

在caffe-master\include\caffe下新建3rdparty文件夾，里面添加hungarian.hpp

下載地址：鏈接: http://pan.baidu.com/s/1o7MTRIi 密碼: xyye

編譯即可

下載數據

http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar

http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar

http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar

在data\VOC0712下解壓VOC文件，變成VOC2007和VOC2012兩個文件夾

在caffe根目錄下新建get_image_size的bat文件

運行bat文件，生成test_name_size.txt文件

復制data\VOC0712的create_data.bat到caffe根目錄下，修改如下root_dir，運行bat文件，在data\VOC0712下生成trainval_lmdb和test_lmdb文件

下載VGG的weight文件https://gist.github.com/weiliu89/2ed6e13bfd5b57cf81d6

下載設置好的訓練和測試的prototxt文件（見上面百度云地址，且這里都已經設置好是CPU了，但是要注意下里面的一些文件路徑與你的路徑是否一致），也可以修改models\VGGNet\VOC0712下的ssd_pascal.py和score_ssd_pascal.py來生成相應的訓練和測試網絡文件及是否采用GPU計算等參數的設置。

在根目錄下新建bat文件

進行訓練，前面的CNN部分是之間用VGG的權重的。

因為只有CPU，所以訓練了一晚上，迭代了2000次，測試下效果

其中test,txt代表是圖像路徑及名稱（類似于HDF5那種形式）具體如下：（最好寫絕對路徑，否則可能讀不到）

原來的項目是預測輸出的是

代表的是類 置信得分 坐標值

為了在opencv顯示顯示其預測結果和保存結果，給ssd_detect.cpp添加一些操作（主要是對image type做了處理，video就沒處理了）

預測的效果如下：

置信度閥值取0.3還能識別兩類，因為訓練的太少了，效果不好，所以有GPU的可以迭代個5萬次左右應該就可以得到很好的結果。實驗室太窮，沒GPU只有訓練下意思意思

版權聲明：本文為博主原創文章，未經博主允許不得轉載。 https://blog.csdn.net/muwu5635/article/details/60874721

總結

以上是生活随笔為你收集整理的微软caffe-SSD的训练和预测（windows cpu）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。