tensorflow
文章平均质量分 78
modelTSS
这个作者很懒,什么都没留下…
展开
-
caffe转tensorflow
https://github.com/ethereon/caffe-tensorflow原创 2020-12-01 15:45:48 · 144 阅读 · 0 评论 -
tensorflow推理过程越来越慢
1. 问题描述tensorflow中多个模型在测试阶段,出现测试速度越来越慢的情况,通过查阅资料发现,由于tensorflow的图是静态图,但是如果直接加在不同的图(即不同的模型),应该都会存在内存中,因此造成了测试速度越来越慢,甚至导致机器卡顿(博主在测试100个模型时,一般测试20个模型左右出现卡顿),因此有必要探究更快的测试速度方法。2. 方法的解决1) 如果采用对于每一个模型均在一个tf.Session()中,会报错。该中方式的代码大致为: for modelNumb.原创 2020-12-01 15:42:24 · 1036 阅读 · 0 评论 -
tensorflow tfrecord制作、读取
使用 TensorFlow 和 TF-Slim 时,对于图片数据集往往需要将数据集转换为 TFRecords 文件.这里根据 TF-Slim 里的flowers的 TFRecords 创建,学习 TFRecords 的创建与读取.以阿里天池竞赛中的服装属性识别的coat_length_labels数据集为例.coat_length_labels = ['Invisible', 'High Waist Length', ...原创 2020-11-25 17:03:32 · 315 阅读 · 0 评论 -
TensorFlow应用.pb文件保存和加载模型方法及相关注意事项
一、.ckpt转.pb用于模型上线.ckpt转.pb主要应用于将训练模型发布上线,.pb模型的跨平台和跨框架性能更好。这里由于在保存.pb模型前需要将模型变量freezing。在应用tensorflow训练模型时,输入数据的batch_size>1,直接保存.pb模型时会在inference阶段出现问题,所以需要从.ckpt转为.pb。在加载.ckpt时可以重新定义输入数据的batch_size=1,以解决该问题。应用步骤主要分为:1.加载.ckpt并且.ckpt转.pb:(1)定义图模型转载 2020-10-22 11:11:23 · 3577 阅读 · 0 评论 -
tf.image.convert_image_dtype坑点:该函数集成了归一化操作!!!
问题:在使用tf.image.convert_image_dtype方法将一个uint类型的tensor转换为float类型时,该方法会自动对数据进行归一化处理,将数据缩放到0-1范围内,如果没有注意到这点,之后在进行网络训练时会发现网络不收敛、不训练。示例:1、读取图像数据并显示可以看到当前的数据类型为int,范围在0-255之间2、使用tf.image.convert_image_dtype转换数据类型为float可以看到经过转换后的数据范围在0-1之间,而且图像数据显示转载 2020-10-22 10:59:26 · 499 阅读 · 0 评论 -
tf.image.resize_images坑点
上上周,我在一个项目上线前对模型进行测试时出现了问题,这个问题困扰了我近两周,终于找到了问题根源,做个简短总结分享给你,希望对大家有帮助。问题描述:线上线下测试结果不一致,且差异很大具体来说,线下测试直接load由ckpt存储的模型,然后使用cv进行数据预处理,然后评估测试集上的准召,一切正常。线上测试时,首先使用tf.image相关函数将预处理写死在模型中,将ckpt模型转为savemodel格式,然后使用tf-serving部署后,发送请求进行线上实测,此时和线下测试结果差异较大。问转载 2020-10-21 15:49:51 · 730 阅读 · 0 评论 -
tensorflow模型部署
https://blog.csdn.net/chongtong/article/details/90474737原创 2020-06-16 10:56:16 · 299 阅读 · 0 评论 -
tensorflow常用概念汇总
tf.GraphKeyshttps://blog.csdn.net/hustqb/article/details/80398934tf.summaryhttps://blog.csdn.net/jasminexjf/article/details/82817912EMAhttps://www.cnblogs.com/cloud-ken/p/7521609.html打印ckp...原创 2020-04-22 22:21:44 · 111 阅读 · 0 评论 -
tensorflow、多GPU、多线程训练VGG19来做cifar-10分类
背景:几天前需要写个多GPU训练的算法模型,翻来覆去在tensorflow的官网上看到cifar-10的官方代码,花了三天时间去掉代码的冗余部分和改写成自己的风格。代码共有6部分构成:1、data_input.py 由于cifar-10官方训练集和验证集都是.bin格式文件,故使用同一份数据读入代码2、network.py 搭建VGG19,返回带weight decay的变量loss和交叉熵之和作...原创 2018-05-29 17:18:57 · 2981 阅读 · 0 评论 -
Ubuntu 16.04 + CUDA 8.0 + cudnn 6.0 + Anaconda 3 + Python 3.6 + tensorflow 1.3
安装https://www.cnblogs.com/xuliangxing/p/7575586.html报错1. import tensorflow 出错libcudnn.so.6: cannot open shared object file: No such file or directoryhttps://blog.csdn.net/qq_29921623/article/details/7...原创 2018-05-02 13:08:06 · 383 阅读 · 0 评论 -
使用localhost访问远程tensorboard
场景与问题tensorflow程序运行在远程服务器,tensorboard启动后访问地址为:0.0.0.0:6006。这样没法用自己机器上的浏览器访问tensorboard。解决办法1 . 连接ssh时,将服务器的6006端口重定向到自己机器上来:ssh -L 16006:127.0.0.1:6006 username@remote_server_ip1其中:16006:127.0.0.1代表自己...转载 2018-05-02 14:42:02 · 850 阅读 · 0 评论