![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow
CCB_307
这个作者很懒,什么都没留下…
展开
-
tensorflowGPU显存占用
问题在毕设使用tensorflow在服务器上跑实验的时候遇到几个问题:tensorflow默认占用所有GPU资源,因此启动就把所有GPU显存给占满。解决办法:使用启用最少的GPU显存来运行程序或者限定显存比例config = tf.ConfigProto() config.gpu_options.allow_growth = True session = tf.Sess...原创 2018-04-15 19:43:36 · 5367 阅读 · 3 评论 -
tensorflow使用自制MNIST数据集训练识别LCD数字模型
项目背景最近老师需要我识别LCD数显仪表的数字,我首先想到的是采用类似车牌分割的办法,先把数字分割出来,再采用神经网络识别数字。为什么不直接将整个LCD图片放进一个网络识别?因为如果这样做的话需要大量的数据,而且可能精度不一定能够得到保证。自制MNIST数据集模仿mnist数据集制作自己的数据集 可以参考上述文章制作数据集 这是我自己生成的LCD数字训练图片,其实我...原创 2018-08-19 23:15:21 · 2254 阅读 · 6 评论 -
tensorflow数据读取——TFRecord
三种办法预加载数据(Preload):在TensorFlow图中定义常量或变量来保存所有数据(仅适用于数据量比较小的情况)。供给数据(Feeding):在TensorFlow程序运行的每一步,让Python代码来供给数据。从文件读取数据:在TensorFlow图的起始,让一个输入管线从文件中读取数。(大数据常用方法)从文件读取数据可以使用TFRecord文件(以二进制存储...原创 2018-09-01 17:55:04 · 368 阅读 · 0 评论