zoukankan      html  css  js  c++  java
  • caffe 生成检测框并绘图

     

    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文件,去掉每行前边的路径,避免重复。

  • 相关阅读:
    JVM-对象的创建
    maven依赖无法下载依赖包,PKIX认证不通过
    Object里面的方法
    Java多线程之volatile与synchronized比较
    java 二叉树的创建 遍历
    设计模式之单例模式
    博客园自定义主题 皮肤
    mysql索引 b+树
    sleep和wait的区别(转)
    final/finally/finalize的区别(转)
  • 原文地址:https://www.cnblogs.com/sddai/p/10221113.html
Copyright © 2011-2022 走看看