具有Imagenet类的TensorFlow对象检测API？

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数据转换为tfrecord for tensorflow对象检测API，你可以参考这个来转换你自己的数据