Added tfdebug

bob/ip/tensorflow_extractor/Extractor.py

@@ -6,6 +6,7 @@
 
 import tensorflow as tf
 import os
+from tensorflow.python import debug as tf_debug
 
 
 class Extractor(object):
@@ -14,7 +15,7 @@ class Extractor(object):
 
     """
 
-    def __init__(self, checkpoint_filename, input_tensor, graph):
+    def __init__(self, checkpoint_filename, input_tensor, graph, debug=False):
         """Loads the tensorflow model
 
         Parameters
@@ -34,17 +35,21 @@ class Extractor(object):
         self.graph = graph
 
         # Initializing the variables of the current graph
-        self.session = tf.Session()
+        self.session = tf.Session()        
         self.session.run(tf.global_variables_initializer())
-
+        
         # Loading the last checkpoint and overwriting the current variables
         saver = tf.train.Saver()
-        
-        if os.path.splitext(checkpoint_filename)[1] == ".meta":
-            saver.restore(self.session, tf.train.latest_checkpoint(os.path.dirname(checkpoint_filename)))
+
+        if os.path.isdir(checkpoint_filename):
+            saver.restore(self.session, tf.train.latest_checkpoint(checkpoint_filename))
         else:
             saver.restore(self.session, checkpoint_filename)
 
+        # Activating the debug
+        if debug:
+            self.session = tf_debug.LocalCLIDebugWrapperSession(self.session)
+
 
     def __del__(self):
         tf.reset_default_graph()
@@ -65,6 +70,5 @@ class Extractor(object):
             The features.
 
         """
-        
         return self.session.run(self.graph, feed_dict={self.input_tensor: data})
 

bob/ip/tensorflow_extractor/__init__.py

 #!/usr/bin/env python
 
 
+from .Extractor import Extractor
+from .InceptionResNet_v1 import InceptionResNet_v1
+
+
 def scratch_network(inputs, end_point="fc1", reuse = False):
 
     import tensorflow as tf
@@ -53,7 +57,6 @@ def download_file(url, out_file):
       with open(out_file, 'wb') as f:
         copyfileobj(response, f)
 
-from .Extractor import Extractor
 
 
 def get_config():