首页 > 解决方案 > Tensorflow 对象检测:使用 .h5 (hdf5) 文件从头开始训练

问题描述

我需要在具有特定配置的 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:我无法找到具有该配置的预训练模型,这就是我需要从头开始训练 cnn 的原因。

标签: pythontensorflowobject-detectionobject-detection-api

解决方案


我的建议是将文件转换为文件,您可以在此处找到如何执行此操作的示例。.hdf5.tfrecord


推荐阅读