摘要: 1.使用tf2onnx工具,把saved model转换为tf的graph def(不带function,也就是tf1的计算图) https://github.com/onnx/tensorflow-onnx/blob/v1.9.3/tf2onnx/tf_loader.py # -*- coding 阅读全文
posted @ 2022-09-29 11:49 灰太狼锅锅 阅读(204) 评论(0) 推荐(0) 编辑
摘要: # -*- coding: utf-8 -*- from tensorflow.python.eager import def_function from tensorflow.python.framework import ops from tensorflow.python.keras.opti 阅读全文
posted @ 2022-09-19 11:45 灰太狼锅锅 阅读(160) 评论(0) 推荐(0) 编辑
摘要: def get_layers_and_variables_from_model(model: tf.keras.Model, scope_name=None): layer_dict = {} if scope_name is not None: base_name = scope_name els 阅读全文
posted @ 2022-09-19 10:52 灰太狼锅锅 阅读(31) 评论(0) 推荐(0) 编辑
摘要: 1.从keras.Model获取output name from tensorflow.python.keras.saving import saving_utils function = saving_utils.trace_model_call(model) concrete_func = fu 阅读全文
posted @ 2022-05-11 16:33 灰太狼锅锅 阅读(172) 评论(0) 推荐(0) 编辑
摘要: 1.auc概念: ①随机挑选一对正负样本,正样本打分值大于负样本的概率 ②roc曲线面积 注意:按照定义①,这里有个问题,打分相同怎么取舍。这个问题可以根据roc曲线绘制方法可知,如果打分值相同的话,roc会有斜线产生,那么计算面积的时候就是梯形面积,等价于打分值相同的时候,取一半的正样本算作打分值 阅读全文
posted @ 2021-10-29 14:17 灰太狼锅锅 阅读(1027) 评论(0) 推荐(0) 编辑
摘要: std::string s = "1.3,1.4,1.5" 字符串中的数字按照逗号分割,把每个数字解析出来. void ParseStr(std::string& s, std::vector<float>& res) { char* pos = const_cast<char*>(s.c_str( 阅读全文
posted @ 2021-10-29 11:45 灰太狼锅锅 阅读(795) 评论(0) 推荐(0) 编辑
摘要: 环境:tensorflow2.2 使用tf.keras.Model.save保存saved_model格式时,默认的input和output比较通用,input_1, input2, output_1,output_2 自定义输入输出名字: import tensorflow as tf sigs 阅读全文
posted @ 2021-10-14 20:38 灰太狼锅锅 阅读(1954) 评论(0) 推荐(0) 编辑
摘要: gdb更友好的打印STL的内容,可以使用gcc提供的printers插件,只需将以下内容写到~/.gdbinit文件中即可: python import sys sys.path.insert(0, '/usr/share/gcc-7/python') from libstdcxx.v6.print 阅读全文
posted @ 2021-08-27 15:05 灰太狼锅锅 阅读(191) 评论(0) 推荐(0) 编辑
摘要: Tensorflow1.13.1中关于nccl的源码位于:tensorflow\core\kernels\nccl_ops.cc;tensorflow\core\nccl\nccl_manager.cc其中核心代码位于nccl_manager.cc文件,入口API为:AddToAllReduce 想 阅读全文
posted @ 2020-03-24 11:05 灰太狼锅锅 阅读(1583) 评论(0) 推荐(0) 编辑
摘要: 在~/.bashrc文件中加入如下内容: alias ll="ls -lh " alias gs="git status " alias gc="git checkout " alias gb="git branch " alias gm="git merge " parse_git_branch( 阅读全文
posted @ 2019-11-15 16:17 灰太狼锅锅 阅读(513) 评论(0) 推荐(0) 编辑