好记性不如烂笔头

在TF1.x中，使用冻结的.pb模型方法比较简单，可以参看：https://editor.youkuaiyun.com/md/?articleId=103403309随着TF2的推进，以后，用TF2的也会越来越多。但是，有时候还是会需要在TF2中使用到TF1.x的模型。对于.pb模型的使用，该如何做呢？直接上codedef wrap_frozen_graph(graph_def, inputs, outputs, print_graph=False): def _imports_graph_def():

Tensorflow中，模型的保存与加载有多种方法。TF1.x保存和加载ckpt格式的模型一般在训练模型的过程中，需要保存模型和权重，常见的方式是保存为ckpt格式的文件，具体的函数如下：tf.train.Saver().save(sess,ckpt_file_path,max_to_keep=4,keep_checkpoint_every_n_hours=2) 再使用类似下面的cod...

参考：https://www.twblogs.net/a/5ca4cadbbd9eee5b1a0713af写这篇笔记的初衷是因为需要将Insightface的MXNET模型转成Tensorflow，而mmdnn上提供的mmconvert命令却没有成功（估计是哪里用错了），故而找其他的文章来实现相同的功能。准备mmdnn环境因为需要将mxnet转成tensorflow，所以，需要在python...

在复用已训练好的模型时，进行需要查看模型网络结构和对应的变量名称，可以使用下面的方法查看。# after load model from file...# print out operations informationops = tf.get_default_graph().get_operations()for op in ops: print(op.name) # prin...

Tensorflow中的GPU分配方法默认情况下 TensorFlow 会使用其所能够使用的所有 GPU，这样，会出现浪费的情况。列出当前设备上的GPU和CPU首先，通过 tf.config.experimental.list_physical_devices，我们可以获得当前主机上某种特定运算设备类型（如 GPU 或 CPU ）的列表，例如，在一台具有 4 块 GPU 和一个 CPU 的工...