Fatal编程技术网
  • tensorflow/
  • 带有Imagenet类的TensorFlow对象检测API？

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

tensorflow neural-network

带有Imagenet类的TensorFlow对象检测API？,tensorflow,neural-network,Tensorflow,Neural Network,在TensorFlow的对象检测API中： 是否可以使用ImageNet classes 1000 classes进行分类？Model Zoo中的4-5个模型都是在COCO数据集上训练的，该数据集只有80个类。是的，您可以将imagenet数据转换为torecord，并训练imagenet检测模型，这与COCO或pascal_voc程序相同。 这里我编写了一个脚本，用于将coco数据转换为tensorflow对象检测API的tfrecord，您可以参考此脚本来转换自己的数据

在TensorFlow的对象检测API中：

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

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