mtcnn人脸检测中将ckpt训练模型转换为 npy保存

from tensorflow.python import pywrap_tensorflow
import numpy as np

#PNet RNet还有ONet模型文件转为npy文件
reader=pywrap_tensorflow.NewCheckpointReader('/mnt/F/lwr/MTCNN/data/MTCNN_model/PNet_landmark/PNet-30')
#reader=pywrap_tensorflow.NewCheckpointReader('F://MTCNN-Tensorflow-master//data//MTCNN_model//RNet_landmark//./RNet-30')

#reader=pywrap_tensorflow.NewCheckpointReader('F://MTCNN-Tensorflow-master//data//MTCNN_model//ONet_landmark//./ONet-22')
var_to_shape_map=reader.get_variable_to_shape_map()
param =[]

for key in var_to_shape_map:
    print ("tensor_name",key)
    param.append(reader.get_tensor(key))

#np.save('ONet.npy',param)
np.save('PNet.npy',param)
# np.save('RNet.npy',param)