1. Tensorflow转换为keras
读取tensorflow权重:参考https://www.jb51.net/article/142183.htm
import tensorflow as tf
checkpoint_path = 'weights/checkpoints/ssd_300_vgg.ckpt'#ssd_300_vgg.ckpt文件夹内存放的是.data和.index
reader = tf.train.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
print("tensor_name: ", key)
# print(reader.get_tensor(key))
#或使用pywrap_tensorflow中的NewCheckpointReader
from tensorflow.python import pywrap_tensorflow
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
#或使用print_tensors_in_checkpoint_file打印ckpt里的内容, 内部使用的是pywrap_tensorflow.NewCheckpointReader
from tensorflow.python.tools.inspect_checkpoint import print_tensors_in_checkpoint_file
print_tensors_in_checkpoint_file(file_name=checkpoint_path, #ckpt文件名字