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

问题描述 投票:0回答:1

我需要通过具有特定配置的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的原因。

python tensorflow object-detection object-detection-api
1个回答
0
投票

我的建议是将.hdf5文件转换为.tfrecord文件,你可以找到如何做这个here的例子。

© www.soinside.com 2019 - 2024. All rights reserved.