
Tensorflow
sooner高
DeepFaceLab作者, Tensorflow & PyTorch contributor,深度学习小学生
展开
-
PyTorch学习笔记(17) ——pytorch的torch.repeat和tf.tile的对比
0. 前言在对比stylegan2的D_stylegan2的时候, 我需要校对我的Minibatch_stddev_layer和作者的tf版本的区别,其中,有一个算子f.tile, 网上已经介绍的很清楚了,将其翻译为pytorch版本的时候其对应的算子没有清晰的解释,因此我将torch.repeat和tf.tile进行了简单比较(下面),基本结论是2者是一致的,可以等效替换.1. 测试比较①...原创 2019-12-29 19:30:24 · 9517 阅读 · 0 评论 -
Tensorflow① 模型转换 ckpt转pb h5转pb
此篇博客重在总结Tensorflow,Keras模型训练的模型文件转换为pb结构的方式,节省寻找转换工具的时间。1. Tensorflow ckpt模型转换pb模型我们在Tensorflow中训练出来的模型一般是ckpt格式的,一个ckpt文件对应有xxx.ckpt.data,xxx.ckpt.meta和xxx.ckpt.index三个内容。而在生产环境中,一般C++只能加载pb的模型,...原创 2018-11-16 11:39:55 · 12412 阅读 · 14 评论 -
Tensorflow② c++接口加载模型推理单张图片
tensorflow的官网上提供很详细的python教程,也确实很好用。但是,应用软件产品大多用c/c++写的。所以,大部分的应用都是用python来训练模型,得到训练好的模型后. 用c++调用tensorflow模型进行推理。通过收集资料,总结了方法.1. 使用python训练模型,并保存模型a.训练并保存模型sess = tf.Session(config=config) #sess...原创 2018-11-06 18:39:54 · 4415 阅读 · 11 评论 -
Tensorflow③ Keras的LSTM和TF的LSTM实现的源码剖析
最近在做可以转成pb模型的RNN/LSTM层的实现细节分析。经过一些分析,发现了在Keras里面常见的keras.layers.LSTM和Tensorflow的tf.contrib.rnn.LSTMCell有一些实现上面的区别。本文将立足于Keras和Tensorflow源码,分别搭建两个简单的一层LSTM的神经网络,验证权重的解析顺序及计算逻辑的正确性。Let’s roll~0. 常见的L...原创 2018-12-19 13:00:18 · 8953 阅读 · 25 评论