SSD中使用自帶工具進行輸出測試結(jié)果以及繪圖

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中目標(biāo)檢測文件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之后如下:

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é)果:

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

友情鏈接更多精彩內(nèi)容