prediction code added

Author: Shashank Tyagi

main_prediction.py

@@ -0,0 +1,29 @@
+import tensorflow as tf
+import os
+from model 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()
+