Step 1 使用训练好的模型检测图片:
build/examples/ssd/ssd_detect.bin models/VGGNet/VOC0712/SSD_300x300/deploy.prototxt /home/daisida/ssd/models/VGGNet/VOC0712/SSD_300x300/VGG_VOC0712_SSD_300x300_iter_120000.caffemodel /home/daisida/ssd/list.txt --out_file tmp/output.txt
参数说明:
ssd_detect.bin:是ssd_detect.cpp的编译文件,类似于caffe中的classifier.bin,主要用于ssd训练模型的检测,相当main函数。
deploy.prototxt:网络结构参数
VGG_VOC0712_SSD_300×300_iter_120000.caffemodel:训练模型
test.txt:测试数据,可以是图片,可以是视频,每一行为图片或视频的路径
--file_type:文件类型,图片用image,视频用video
--out_file:检测结果输出文件
--condidence_threshold:阈值参数,这里设置为0.5,参数范围(0,1)
Step 2 结果可视化:
保存结果前需要将output.txt中的路径进行修改,去掉每行前边的路径,避免重复。
python examples/ssd/plot_detections.py tmp/output.txt /home/daisida/data/testset --labelmap-file data/VOC0712/labelmap_voc.prototxt --save-dir tmp/
报错:
/home/daisida/data/testset///home/daisida/data/testset/3.jpg does not exist
/home/daisida/data/testset///home/daisida/data/testset/9.jpg does not exist
/home/daisida/data/testset///home/daisida/data/testset/7.jpg does not exist
/home/daisida/data/testset///home/daisida/data/testset/2.jpg does not exist
/home/daisida/data/testset///home/daisida/data/testset/8.jpg does not exist
解决方案:
修改output.txt文件,去掉每行前边的路径,避免重复。