(self, checkpoint_dir, image_width, image_height)
Setup model for inference.

        Build computation graph, initialize variables, and load checkpoint.
Setup model for inference.
Setup model for inference .
def setup_for_inference(self, checkpoint_dir, image_width, image_height):
        """Setup model for inference.

        Build computation graph, initialize variables, and load checkpoint.
        """
        self.image_width = image_width
        self.image_height = image_height
        # Figure out image dimension
        self._build_feed_model()
        ginit_op = tf.global_variables_initializer()
        linit_op = tf.local_variables_initializer()
        self.sess.run([ginit_op, linit_op])

        if self.load(checkpoint_dir):
            print("[*] Load Success")
        else:
            raise RuntimeError('MagNet: Failed to load checkpoint file.')
        self.is_graph_built = True
def setup_for_inference ( self , checkpoint_dir , image_width , image_height ) : self . image_width = image_width self . image_height = image_height # Figure out image dimension self . _build_feed_model ( ) ginit_op = tf . global_variables_initializer ( ) linit_op = tf . local_variables_initializer ( ) self . sess . run ( [ ginit_op , linit_op ] ) if self . load ( checkpoint_dir ) : print ( "[*] Load Success" ) else : raise RuntimeError ( 'MagNet: Failed to load checkpoint file.' ) self . is_graph_built = True