![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tf2.0
qq_41422774
Python萌新
展开
-
TF2.0里面的tfrecord
记录一下最近遇到的问题TF2里面的tfrecord数据格式制作与加载制作没什么好讲的,代码如下:def _make_tfrecords(doc_path, save_dir, train_or_test): tfrecords_path = save_dir +'\\' +train_or_test +'.tfrecord' if os.path.exists(tfre...原创 2019-12-25 21:58:24 · 1076 阅读 · 0 评论 -
tensorflow2.0显存设置
遇到一个问题:新买显卡显存8G但是tensorflow运行的时候界面显示只有约6.3G的显存可用,如下图:即限制了我的显存,具体原因为什么我也不知道,但原来的显存小一些的显卡就没有这个问题。目前的解决办法是:官方文档解决然后对应的中文博客总结一下,就是下面的两个办法:设置每次使用时按需申请# 获取所有GPU组成listgpus = tf.config.experimenta...原创 2019-12-22 21:38:03 · 4354 阅读 · 1 评论 -
tensorflow2.0读取修改保存图片
import tensorflow as tfimport numpy as npfrom matplotlib import pyplot as pltimg_path = r'C:\Users\Nolan\Desktop\py\TF2_0\Dogs\test\chihuahua\n02085620_9357.jpg'# 读入图片img = tf.io.read_file(img_p...原创 2019-11-07 20:04:41 · 2479 阅读 · 1 评论