我需要通过具有特定配置的COCO数据集从头开始训练CNN:https://github.com/tensorflow/models/blob/master/research/object_detection/samples/configs/embedded_ssd_mobilenet_v1_coco.config
因此,我安装了TF Object Detection API,并下载了COCO数据集。但是,数据集的扩展名为.h5。是否可以使用这种文件运行培训,或者我是否需要在图像中进行转换?如果可能,该命令会是什么?
PS:我无法找到使用config的预训练模型,这就是我需要从头开始训练cnn的原因。
我的建议是将.hdf5
文件转换为.tfrecord
文件,你可以找到如何做这个here的例子。