预先训练的CNN(tensorflow / darknet / caffe)权重,仅用于人/车辆检测

时间:2019-09-26 11:08:09

标签: python tensorflow caffe yolo

我正在使用Tensorflow detection model zoo中的张量流的预训练权重,该权重主要在COCO数据集上进行训练,该数据集涵盖了包括人类在内的大约80个不同类别,因此使用这些模型肯定会导致更高的计算量需要进行权衡,因此在这种情况下,是否有任何公开开放的预训练权重仅关注一个类别,无论是人类还是车辆。

如果没有可用的模型,那么如何微调或自定义这些现有模型,如

ssd_inception_v2_coco_2018_01_28 ”,在 mAP为32且计算效率也很好的情况下,表现良好,因此如何将此类模型用于仅检测人类而不检测其他任何人类对象。

1 个答案:

答案 0 :(得分:1)