常用工具
关注
分享
扫一扫
编写一些常用的工具脚本~
ciky奇
这个作者很懒，什么都没留下…
展开
-
tensorflow1.X版本中打印训练好模型各个节点名，并保存在txt，根据输入输出节点保存‘.pb’文件
tensorflow模型训练好后有几个文件，其中".ckpt"是模型参数数据，“.meta”存有网络结构；如何打印显示网络中各个节点的名称（op），代码如下：import tensorflow as tfckpt_path = './model.ckpt'saver = tf.train.import_meta_graph(ckpt_path+'.meta',clear_devic...原创 2020-03-26 15:18:19 · 3448 阅读 · 7 评论 -
tensorflow中将标注文件写到train.txt, test.txt,trainval.txt中
tensorflow中对于标注好的文件以及图片在训练之前会将其打散然后分成训练用的，测试用的，验证用的；以下代码就是干这事~import osimport random trainval_percent = 0.66 # 标注文件中训练验证所占的比例 train_percent = 0.5 # 训练验证中训练所...原创 2020-03-26 14:14:38 · 2574 阅读 · 4 评论