简体   繁体   中英

Tensorflow Object Detection: training from scratch using a .h5 (hdf5) file

I need to train from scratch a CNN over a COCO dataset with a specific configuration: https://github.com/tensorflow/models/blob/master/research/object_detection/samples/configs/embedded_ssd_mobilenet_v1_coco.config

Thus, I installed TF Object Detection API and I downloaded the COCO dataset. However the dataset is in .h5 extension. Is it possible to run the training with this kind of file or do I need to convert it in images in someway? If that is possible, what would the command be?

PS: I was not able to find a pre-trained model with that config , this is why I need to train a cnn from scratch.

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

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM