tensorflow saver.restore()

加载pretrained_model模型

-需要注意的是,tensorflow提供了下列函数,可以将pretrained_model设为文件夹,自动检查并加载上次save的ckpt文件。
- 下面是facenet训练时的加载上次模型的代码。需要在源代码中修改


saver = tf.train.Saver(tf.trainable_variables(), max_to_keep=3)
ckpt = tf.train.get_checkpoint_state(args.pretrained_model)
print('Restoring pretrained model: %s' % args.pretrained_model)
saver.restore(sess, ckpt.model_checkpoint_path)

你可能感兴趣的:(tensorflow,人脸识别)