使用Imagenet类的TensorFlow对象检测应用编程接口？

Question

在TensorFlow的对象检测API中：

https://github.com/tensorflow/models/tree/master/object_detection

是否可以使用ImageNet类别(1000个类别)进行分类？模型动物园中的4-5个模型都是在COCO数据集上训练的，COCO数据集只有80个类。

Answer 1

可以，您可以将imagenet数据转换为torecord，并训练imagenet检测模型，它与coco或pascal_voc程序相同。在这里，我编写了一个脚本tensorflow_object_detection_create_coco_tfrecord来将coco数据转换为tensorflow对象检测API的tfrecord，您可以参考此脚本来转换您自己的数据。