帶有Imagenet類的TensorFlow對象檢測API？

Question

在TensorFlow的物體檢測API：

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

是否有可能使用ImageNet類（1000級）進行分類？模型動物園中的4-5個模型都在COCO數據集上進行培訓，該數據集只有80個類。

Answer 1

是的，您可以將imagenet數據轉換爲torecord，並訓練imagenet檢測模型，這與coco或pascal_voc過程相同。 這裏我寫一個腳本tensorflow_object_detection_create_coco_tfrecord來將coco數據轉換爲tfrecord for tensorflow對象檢測API，你可以參考這個來轉換你自己的數據