在预训练的模型中不需要一些现有的课程

时间:2018-03-14 05:49:22

标签: tensorflow object-detection

Tendorflow对象检测。

我知道你可以在这里用现有的预训练模型训练新课程:

https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md

但我想只检测一个人'和' car',这两个类已经存在了,还有 培训不是必要的。

但是,' faster_rcnn_inception_v2_coco'这样的模型,它检测到90个类,对我来说运行速度太慢。

有没有办法减少类,所以检测运行得更快?

PS:我已经将COCO数据集过滤到只有“人”的数据集。然后按照我用Google搜索的常规步骤,我得到了不好的训练结果。我只能成功地检测到了人物的原因。

1 个答案:

答案 0 :(得分:1)

解决了我自己。如果你使用了'#39;或者' car'在coco中,在预先训练的模型中id为1和3,并且无法更改。

在.config中,设置' num_classes' = 3,尽管您的类数为2。

3是适合汽车' -id。

(例如,如果您选择2个班级,' person' = 1,' truck' = 8,您的' num_classes'应为8)

另外,正确选择您的纪元,您可以看到:http://neuralnetworksanddeeplearning.com/chap3.html