tensorflow一些代码

load多个图：

class ImportGraph():
    """  Importing and running isolated TF graph """
    def __init__(self, loc):
        # Create local graph and use it in the session
        self.graph = tf.Graph()
        config = tf.ConfigProto(log_device_placement=False)
        config.gpu_options.allow_growth = True
        self.sess = tf.Session(graph=self.graph, config=config)
        with self.graph.as_default():
            # Import saved model from location 'loc' into local graph
            saver = tf.train.import_meta_graph(loc + '.meta',
                                               clear_devices=True)
            saver.restore(self.sess, loc)
            # There are TWO options how to get activation operation:
              # FROM SAVED COLLECTION:
            self.logits = self.graph.get_operation_by_name('proj/Reshape_1').outputs[0]
            # self.activation = tf.get_collection('activation')[0]
              # BY NAME:
            # self.activation = self.graph.get_operation_by_name('activation_opt').outputs[0]

    def run(self, fd):
        """ Running the activation operation previously imported """
        return self.sess.run([self.logits],
                             feed_dict=fd)

import tensorflow as tf
class Model:
    def __init__(self,param):
        self.param = param

        # create & build graph
        self.graph = tf.Graph()
        self.build_graph()

        # create session
        config = tf.ConfigProto()
        config.gpu_options.allow_growth = True
        gpu_num = random.choice(cuda_gpu_count())
        config.gpu_options.visible_device_list= str(gpu_num)
        self.sess = tf.Session(config=config,graph=self.graph)

    def build_graph(self):
        with self.graph.as_default():
            ...
    def __del__(self):
        # explicitly collect resources by closing and deleting session and graph
        self.sess.close()
        del self.sess
        del self.graph
        del self.param

    # train models and return the test accuracy
    def train_test(self,train_data,train_label,test_data,test_label):
        ...

---

https://blog.csdn.net/silent56_th/article/details/81415940