深度学习tensorflow实战笔记(2)图像转换成tfrecords和读取

     上一篇博客介绍了如何把自己的一般数据放进去txt文件并读取训练全连接神经网络(多层感知机),但是如何把自己的图像数据投入到神经网络中做训练呢,当然,可以把图像写入到txt文件中,然后做reshape,这种方法效率太低,操作复杂,并且内存占用率太高。所以本篇博客介绍如何把自己的图像数据转成tensorflow平台标注的tfrecords格式,并读取tfrecords数据。下一篇博客介绍如何使用tfrecords数据用作CNN的训练。

1、准备数据

      首选将自己的图像数据分类分别放在不同的文件夹下,比如新建data文件夹,data文件夹下分别存放up和low文件夹,up和low文件夹下存放对应的图像数据。也可以把up和low文件夹换成0和1。根据自己数据类别,自己设定。如图所示

      



     以上三张图片注意看目录。这样数据就准备好了。

2、将图像数据转换成tfrecords
      直接上代码,代码中比较重要的部分我都做了注释。
import os
import tensorflow as tf 
f
  • 2
    点赞
  • 49
    收藏
    觉得还不错? 一键收藏
  • 13
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值