本文介绍了如何从TensorFlow-Slim库获取MobileNet_v2_1.0_224预训练模型，并详细列出了包含的文件及其作用。在Test6_mobilenet/pretain_model目录下存放了模型的重要文件，通过指定的Python代码可以生成新的预训练模型权重文件。
从TensorFlow-Slim image classification model library中找到我们要使用的MobileNet_v2_1.0_224^*（1.0-阿尔法值，244-输入图片shape），下载mobilenet_v2_1.0_224.tgz文件，其中内含：
- mobilenet_v2_1.0_224.ckpt.data-00000-of-00001
- mobilenet_v2_1.0_224.ckpt.index
- mobilenet_v2_1.0_224.ckpt.meta
- mobilenet_v2_1.0_224.tflite
- mobilenet_v2_1.0_224_eval.pbtxt
- mobilenet_v2_1.0_224_frozen.pb
- mobilenet_v2_1.0_224_info.txt
mobilenet_v2_1.0_224.ckpt.data-00000-of-00001存储模型权重，mobilenet_v2_1.0_224.ckpt.meta存储模型图的流程，mobilenet_v2_1.0_224.ckpt.index存储模型结构的变量与参数间的索引对应关系，mobilenet_v2_1.0_224_frozen.pb冻结的PB文件，mobilenet_v2_1.0_224.tflite冻结的tflite文件。
在Test6_mobilenet下新建目录pretain_model，将以下三个文件放入其中
- mobilenet_v2_1.0_224.ckpt.data-00000-of-00001
- mobilenet_v2_1.0_224.ckpt.meta
- mobilenet_v2_1.0_224.ckpt.index
生成预训练模型代码，运行后在当前路径下生成两个
最低0.47元/天 解锁文章
扫一扫
2181
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
