TensorFlow其他
TensorFlow其他
业余狙击手19
图像算法菜鸟,点滴积累成就硕果累累。
展开
-
TensorFlow-softmax_cross_entropy()、binary_cross_entropy()、sigmoid_cross_entropy()之间的区别与联系
损失函数softmax_cross_entropy、binary_cross_entropy、sigmoid_cross_entropy之间的区别与联系原创 2020-02-22 10:05:32 · 689 阅读 · 0 评论 -
TensorFlow-tf.concat()函数解析
眼看千遍,不如手动一遍,看了原文再手动整理一遍,代码实际操作一遍,加深理解。相当于高中时做的笔记了。concat()是将tensor沿着指定维度连接起来。其中tensorflow1.3版中是这样定义的:concat(values,axis,name='concat')一、对于2维来说,0表示行,1表示列t1 = [[1, 2, 3], [4, 5, 6]]t2 = [[7, ...原创 2019-08-30 19:12:59 · 2306 阅读 · 0 评论 -
TensorFlow-tf.assign()函数解析
tf.assign(A, new_number): 这个函数的功能主要是把A的值变为new_number#import tensorflow as tf#创建一个变量,初始化为标量0state=tf.Variable(0,name='counter')#创建一个op,其作用是使state增加1one=tf.constant(1)new_value=tf.add(state,o...原创 2018-10-23 11:49:06 · 2311 阅读 · 0 评论 -
TensorFlow-tf.nn.dynamic_rnn函数解析
眼看千遍,不如手动一遍,看了原文再手动整理一遍,代码实际操作一遍,加深理解。相当于高中时做的笔记了。tf.nn.dynamic_rnn 函数是tensorflow封装的用来实现递归神经网络(RNN)的函数,本文会重点讨论一下tf.nn.dynamic_rnn 函数的参数及返回值。首先来看一下该函数定义:tf.nn.dynamic_rnn( cell, inputs,...原创 2019-08-30 19:12:35 · 2890 阅读 · 0 评论 -
TensorFlow-tf.nn.embedding_lookup()函数解析
眼看千遍,不如手动一遍,看了原文再手动整理一遍,代码实际操作一遍,加深理解。相当于高中时做的笔记了。tf.nn.embedding_lookup的用法主要是选取一个张量里面索引对应的元素原型:tf.nn.embedding_lookup(params, ids, partition_strategy='mod', name=None, validate_indices=True, m...原创 2019-08-30 19:13:14 · 2425 阅读 · 0 评论 -
ModuleNotFoundError: No module named 'utils.cython_bbox'
源码地址:https://github.com/lijianaiml/tf-faster-rcnn-windows从源码中找到“tf-faster-rcnn-windows-master\lib\setup.py”,注释掉66行的# CUDA = locate_cuda()用pycharm打开,右击,选择:open in terminal先输入以下命令执行:python se...原创 2019-12-07 10:10:09 · 15445 阅读 · 0 评论 -
TensorFlow-最基本的seq2seq写法
以此为核心,加上各种改进技巧,造就强大的seq2seq!Action!import tensorflow as tfclass Seq2seq(object): def __init__(self, config, w2i_target): self.seq_inputs = tf.placeholder(shape=(config.batch_size, No...原创 2019-09-07 16:15:38 · 2064 阅读 · 0 评论 -
TensorFlow-重写计算图的方式保存pb,Python调用pb没问题,OpenCV调用pb,不能加leaky_relu和BN层
将网络模型(图+权值)保存为.pb文件 write.py采用的思路是重写计算图的方法# -*- coding: utf-8 -*-from __future__ import absolute_import, unicode_literalsfrom tensorflow.examples.tutorials.mnist import input_dataimport tens...原创 2019-08-22 10:15:30 · 2537 阅读 · 4 评论 -
使用TensorFlow、CNN实现MNIST手写字体识别,并利用tensorboard实现可视化
CNN: 卷积层1([5,5,1,32]),池化层1(2*2); 卷积层2([5,5,32,64]),池化层2(2*2); 全连接层1(7*7*64,1024); 全连接层2(1024*10);import tensorflow as tffrom tensorflow.examples.tutorials.mnist import input_data#载入数据集mnist =...原创 2018-09-03 17:19:35 · 4432 阅读 · 0 评论 -
TensorFlow保存训练模型为pb文件,读取pb恢复模型,C++调用pb模型
这个问题我是看了网上好多博客,自己也踩了好多坑才搞定的,看我这一篇文章,可能不能完全解决你们的问题,但是希望对你们有所帮助!python线上训练模型比较方便,有时候想要在C++环境下调用python训练好的模型,应该怎么做呢?折腾了好几天,终于搞定了! 一,第一步,训练保存模型 贴自己源代码太复杂,贴个简化版,表明主要意思就行,别人做的东西不可能和你的完全一样,需要在理解别人意思的基础上,...原创 2018-08-22 09:07:13 · 33051 阅读 · 50 评论 -
Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 (python3.6)
输入以下代码,会出现红色提示:import tensorflow as tfa=tf.constant([[1.0,2.0]])b=tf.constant([[3.0],[4.0]])result=tf.matmul(a,b)print(result)with tf.Session() as sess: print (sess.run(result))解决方式是...原创 2018-07-22 11:32:22 · 2375 阅读 · 0 评论