Tensorflow对象检测:使用.h5(hdf5)文件从头开始训练

时间:2018-08-03 15:52:47

标签: python tensorflow object-detection object-detection-api

我需要从头开始通过具有特定配置的https://github.com/tensorflow/models/blob/master/research/object_detection/samples/configs/embedded_ssd_mobilenet_v1_coco.config

在COCO数据集上训练CNN

因此,我安装了TF Object Detection API,并下载了COCO数据集。但是,数据集的扩展名为.h5。 是否可以使用此类文件进行培训,或者我是否需要以某种方式将其转换为图像?如果可能的话,命令将是什么?

PS:我无法使用该config找到预先训练的模型,这就是为什么我需要从头开始训练cnn的原因。

1 个答案:

答案 0 :(得分:0)

我的建议是将.hdf5文件转换.tfrecord文件,您可以找到有关如何here的示例。