注册

SSD模型如何修改文件显示指定数目的类别？

大家好，
  我正在caffe上运行SSD模型，当运行ssd_pascal_video.py时显示20个类的文本信息，如果我只想让模型只显示2个类的信息（比如车和人），请问如何修改文件？谢谢！
2017-01-08 添加评论
分享

没有找到相关结果

    已邀请:

    jiangqr

    赞同来自: xinmiao

    类别信息是在此行定义的：label_map_file = "data/VOC0712/labelmap_voc.prototxt"
    如果要检测结果只出两个类别，可以重新训练只有两类的模型，或者修改c++代码不显示其它类，
    直接修改prototxt文件估计跑不起来。
    2017-01-08 1 0
    分享

    dragon518

    赞同来自:

    谢谢,我试试
    2017-01-08 0 0
    分享
    为什么被折叠? 0 个回复被折叠

    要回复问题请先登录或注册

    Copyright © 2021