有没有办法限制现有的TensorFlow对象检测模型？

Question

在GitHub上的tensorflow / models repo上，他们提供了五个用于物体检测的预训练模型。

这些模型在COCO数据集上进行训练，可以识别90个不同的对象。

我需要一个模型来检测人，而不是其他任何东西。我可以修改代码只打印人物上的标签，但它仍然会查找其他89个对象，这比寻找一个对象需要更多的时间。

我可以训练自己的模型，但我宁愿能够使用预先训练过的模型，而不是花费大量时间训练我自己的模型。

有没有办法，通过修改模型文件，或TensorFlow或Object Detection API代码，所以它只查找单个对象？

Answer 1

要么对行人进行微调，只需要几千步就可以了（要训练的小数据集就足够了），或者在标签定义文件（.pbtx文件）中查找，搜索人物标签，并且和其他人一起做你想做的事。