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

大家好,
  我正在caffe上运行SSD模型,当运行ssd_pascal_video.py时显示20个类的文本信息,如果我只想让模型只显示2个类的信息(比如车和人),请问如何修改文件?谢谢!
已邀请:

jiangqr

赞同来自: 辛淼

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

dragon518

赞同来自:

谢谢,我试试

要回复问题请先登录注册