convert_ckpt.py File Reference

Namespaces
	convert_ckpt

Functions
def	convert_ckpt.get_weights (trainable_ops)
def	convert_ckpt.get_activations (vis_ops)
def	convert_ckpt.get_vis2_layers (vis_ops, weight_ops, weight_values)

Variables
	convert_ckpt.b_test = np.load('dataset/batch_test.npy')
	convert_ckpt.x_test = b_test[:, 1:]
	convert_ckpt.y_test = b_test[:, 0]
	convert_ckpt.dataset_test = tf.data.Dataset.from_tensor_slices((x_test, y_test)).repeat().shuffle(buffer_size=500).batch(1)
	convert_ckpt.new_saver = tf.train.import_meta_graph('ckpt/model.ckpt-{}.meta'.format(i))
	convert_ckpt.dataset_init_op = sess.graph.get_operation_by_name('dataset_init')
	convert_ckpt.logits = model.inference(features)
	convert_ckpt.vis_ops = tf.get_collection('VisOps')
	convert_ckpt.train_ops = tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES)
def	convert_ckpt.weight_ops = get_weights(train_ops)
	convert_ckpt.weight_values = sess.run(weight_ops)
	convert_ckpt.activation_values = sess.run(get_activations(vis_ops))
def	convert_ckpt.json_base = get_vis2_layers(vis_ops, weight_ops, weight_values)