![](https://img-blog.csdnimg.cn/20201014180756738.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow
SrCMpunk
现在才开始写博客,有点晚了。
展开
-
[tensorflow] 多维矩阵的乘法
[tensorflow] 多维矩阵的乘法原创 2022-08-19 11:00:01 · 497 阅读 · 0 评论 -
tensorflow 图不是拓扑结构问题
问题:topological sort failed with message: The graph couldn’t be sorted in topological order.背景:我确认自己的建立的图没有环,但是运行的时候tensorflow会报graph中存在非拓扑结构原因:使用n卷积,每个卷积的输入都是相同的数据data,这样会得到n个卷积结果,然后按axis=3的维度进行合并,这个时候会报错。解决方案:不要在最后将n个卷积结果一次性的合并,而是要在每次卷积后与之前的卷积进行合并。原创 2020-10-12 15:19:46 · 801 阅读 · 0 评论 -
Partial Convolution implement Tensorflow 1.X version
inpainting里面重要的一个工作是Nvidia的Partial Convolution部分卷积,该卷积包含两个分支,一个是更新mask图像,另一个则是根据mask图进行部分的卷积。下面是他的1.x版本的实现代码如下:def pc_conv(x, mask, cnum, ksize, stride, test ,name='pconv'): b, h, w, c = int(x.s...原创 2020-03-09 14:03:28 · 552 阅读 · 0 评论 -
The BatchNorm layer convertion to caffe model from tensorflow
caffe must set the batchnorm scale as 1, which is as follows.net.params[‘bn1’][2].data[…] = 1Mean value is [0,0,0,0…,0] and variance is [1,1,1,1,…,1]The gamma and beta value are from tensorflow mo...原创 2020-01-07 16:41:01 · 160 阅读 · 0 评论 -
Count to tensorflow model FLOPs and trainable params
Count to tensorflow model FLOPs and trainable paramsStep1:Convert your tensorflow checkpoint model to PB formatStep2:Call the function as followsimport tensorflow as tffrom tensorflow.python.fra...原创 2020-01-03 14:39:07 · 581 阅读 · 0 评论 -
tensorflow gauss blur 高斯模糊处理
tensorflow的高斯模糊函数如下import tensorflow as tfimport numpy as npfrom scipy.ndimage.filters import gaussian_filterfrom ops import concatdef gauss_kernel_fixed(sigma, N): # Non-Adaptive kernel si...原创 2019-09-11 17:00:36 · 4034 阅读 · 0 评论 -
tensorflow fake quantization 伪量化训练-单GPU和多GPU
tensorflow fake quantization 伪量化训练tensorflow伪量化训练简单说就是用float类型的模拟int类型的运算。在fake quantization训练的过程中,尽量使float类型的计算精度接近int类型的精度。fake quantization 需要在计算图中添加一个伪量化的节点,才能进行伪量化训练,同时该方法的训练出来的模型需要使用,对应的伪量化转pb代...原创 2019-09-02 09:20:48 · 2139 阅读 · 0 评论 -
tensorfow checkpoint 转pb 模型 conv2d的输出节点名字是BiasAdd
constant_graph = graph_util.convert_variables_to_constants(sess, sess.graph_def, ['*/BiasAdd'])为自己定义的name scope。如果使用tf.layers.conv2d实现的卷积操作,要将pb模型的输出结果为该卷积的结果,需要指定为 /BiasAdd而不是/Conv2D...原创 2019-08-06 11:11:52 · 481 阅读 · 0 评论 -
How to process Conv weight during the model trainning
这里写自定义目录标题Tensorflow如何在训练的时候对卷积的权重进行一些特殊处理步骤如下整体代码Tensorflow如何在训练的时候对卷积的权重进行一些特殊处理比如对卷积的权重进行求均值,取最大值等等自定义的操作。步骤如下获取想要操作的的卷积核权重,保存到一个list中申请一个同该卷核相同shape的占位符;使用tf.assign()创建一个更新卷积核权重的操作op;整体代码...原创 2019-07-22 16:40:27 · 166 阅读 · 0 评论 -
使用tf.layers.conv2d()转PB模型时记得加Conv2D后缀
tf.layers.conv2d()会自动添加Conv2D例如我们这样设置tf.layers.conv2d( , name='output')我们将checkpoint转换成pb模型时,需要制定一个输出节点。使用layers这个包,如name为output,如果直接设定该节点为输出节点,则会报错。原因:tensorflow的layers会自动的name的后面加上Conv2D下级scope...原创 2019-07-29 16:31:54 · 261 阅读 · 0 评论 -
tensorflow用tf.nn.conv2d_transpose实现tf.layers.conv2d_transpose
用nn的操作实现layers的操作def deconv(x, cnum, kernel_size = 3, scale_size = 2,stride = 2, name='upsample', padding='SAME', ): b, w, h, c = x.shape[0], x.shape[1], x.shape[2], x.shape[3] b, w, h, c = i...原创 2019-07-29 14:41:42 · 882 阅读 · 0 评论 -
Tensrfow GAN Discriminator 如何使用hinge loss训练
hinge loss核心点:网络的输出要确保是[-1,1]范围之前一直用cross entrype loss这一点没有台注意,所以之前一直没写对!hinge loss 核心代码 def Hinge_loss(pos, neg, name='Hinge_loss'): with tf.variable_scope(name): d_loss ...原创 2019-04-24 10:17:51 · 4573 阅读 · 4 评论