main_prediction.py

import tensorflow as tf
import os
from model_prediction import *


if not os.path.exists('../logs'):
	os.makedirs('../logs')

if not os.path.exists('../checkpoint'):
	os.makedirs('../checkpoint')

if not os.path.exists('../best_checkpoint'):
	os.makedirs('../best_checkpoint')

map(os.unlink, (os.path.join( '../logs',f) for f in os.listdir('../logs')) )

net = HyperFace(True, tf_record_file_path='./aflw_train_small_check.tfrecords',model_save_path='../checkpoint/',best_model_save_path='../best_checkpoint/',
restore_model_path='../full_best_checkpoint/')

with tf.Session() as sess:
		print 'Building Graph...'
		net.build_network(sess)
		print 'Graph Built!'
		# net.print_variables()
		# net.load_weights('/Users/shashank/TensorFlow/SPN/weights/')
		net.predict()
		# net.train()