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

【问题标题】：TensorFlow Object Detection API with Imagenet classes?带有 Imagenet 类的 TensorFlow 对象检测 API？
【发布时间】：2017-12-23 21:47:55
【问题描述】：

在 TensorFlow 的对象检测 API 中：

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

是否可以使用 ImageNet 类（1000 个类）进行分类？ Model Zoo 中的 4-5 个模型都是在 COCO 数据集上训练的，COCO 数据集只有 80 个类。

【问题讨论】：

标签： tensorflow neural-network

【解决方案1】：

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

【讨论】：