从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