![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
TensorFlow
MissShihongHowRU
这个作者很懒,什么都没留下…
展开
-
TensorFlow Dataset的使用
内容摘自 --《TensorFlow 实战Google深度学习框架》 第二版1.4 版本以后,数据集框架从tf.contrib.data 迁移到 tf.data,它被TensorFlow推荐作为输入数据的首选框架。由于训练数据通常无法全部写入内存中,从数据集中读取数据时需要使用一个迭代器(iterator)按顺序进行读取。与队列相似,数据集也是计算图上的一个节点。input_data =...原创 2019-08-15 19:58:49 · 303 阅读 · 0 评论 -
TensorFlow collection的概念
内容摘自《TensorFlow实战Google深度学习框架》 第二版集合(collection)有效整理TenserFlow程序中的资源也时计算图的一个重要功能。在一个计算图中,可以通过集合(collection) 来管理不同个类别的资源。 tf.add_to_collection 函数可以将资源加入一个或多个集合中,然后通过 tf.get_collection 获取一个集合里面的所有资源...原创 2019-08-10 21:52:27 · 378 阅读 · 0 评论 -
TensorFlow 计算加速(一)
TensorFlow 计算加速内容摘自《TensorFlow实战Google深度学习框架》 第二版1. TensorFlow使用GPUTensorFlow程序可以通过tf.device函数来通过名称指定运行每一个操作的设备,这个设备可是是本地的GPU或CPU,也可以是一台远程的服务器。在默认情况下,就算及其有多个CPU,TensorFlow也不会区分他们,所有的CPU都使用/cpu:...原创 2019-08-10 14:52:07 · 743 阅读 · 0 评论 -
在服务器的docker中部署深度学习模型(flask框架)
目录服务端客户端目标: 我需要在测试polygonrnn++的预测效果,并把模型部署在服务器上。服务端首先在服务端搭建配置好环境的docker。 新建容器时有几个注意点:使用nvidia-docker 或者设置 --runtime 参数,否则 import tensorflow (gpu版) 时会报错: file too short使用 -p 参数设置多个端口映射, :左边是服务器端口...原创 2019-11-29 18:30:11 · 1579 阅读 · 0 评论 -
tf.segment_sum()_ tf.while_loop()_ tf.clip_by_value()_ tf.tile()_ tf.unique_with_counts()
这几天 在阅读大项目LaneNet的损失函数计算部分时,遇到了许多未曾一见的tensorflow的函数,上网查找资料后做一个学习笔记。分段求和 tf.segment_sum()1.14之后将它移到了math模块下tf.math.segment_sum( data, segment_ids, name=None)outputi=∑jdatajoutput_i =...原创 2019-11-13 11:17:42 · 464 阅读 · 1 评论