1.使用一下命令進行輸出測試結(jié)果:
./build/examples/ssd/ssd_detect.bin /root/caffe-ssd/models/VGGNet/MyDataSet/SSD_300x300/deploy.prototxt /root/caffe-ssd/models/VGGNet/MyDataSet/SSD_300x300/VGG_MyDataSet_SSD_300x300_iter_40000.caffemodel /root/data/MyDataSet/SSD300x300/test.txt --out_file?/root/Results/MyDataSet/SSD300x300/output.txt --confidence_threshold 0.6?
參數(shù)解讀:
1 ./build/examples/ssd/ssd_detect.bin??ssd_detect.bin是ssd中目標檢測文件ssd_detect.cpp的編譯文件邻吞,可以直接使用组题。(關(guān)鍵)
2?/root/caffe-ssd/models/VGGNet/MyDataSet/SSD_300x300/deploy.prototxt? 這個是你的網(wǎng)絡(luò)結(jié)構(gòu)參數(shù),在訓(xùn)練SSD時候會自動生成吃衅。
3?/root/caffe-ssd/models/VGGNet/MyDataSet/SSD_300x300/VGG_MyDataSet_SSD_300x300_iter_40000.caffemodel? ?這里是訓(xùn)練模型
4?/root/data/MyDataSet/SSD300x300/test.txt? ?這里是測試數(shù)據(jù)文件
5?--out_file?/root/Results/MyDataSet/SSD300x300/output.txt? ?這里是輸出結(jié)果
6.--confidence_threshold 0.5? ?置信度閾值設(shè)置往踢,只有大于這個值得框才會輸出。
針對SSD中test.txt文件中沒有路徑信息徘层,只有圖片編號會導(dǎo)致后面顯示圖片找不到的錯誤的解決辦法:
使用excel進行批量添加路徑信息:即可
之后復(fù)制會test.txt之后如下:
下面是output_0.5.txt的文件內(nèi)容:
2.根據(jù)結(jié)果繪圖:
python examples/ssd/plot_detections.py /root/Results/MyDataSet/SSD300x300/output_0.5.txt /root/data/VOCdevkit --labelmap-file data/MyDataSet/labelmap_voc.prototxt --save-dir /root/Results/MyDataSet/SSD300x300/ssd_result_0.5
參數(shù)解析:
1?python examples/ssd/plot_detections.py
2?/root/Results/MyDataSet/SSD300x300/output_0.5.txt
3 /root/data/VOCdevkit
4 --labelmap-file data/MyDataSet/labelmap_voc.prototxt
5?--save-dir /root/Results/MyDataSet/SSD300x300/ssd_result_0.5
最后在相應(yīng)目錄下即可得到最終結(jié)果: