tensorflow生成多个tfrecord文件实例


Posted in Python onFebruary 17, 2020

我就废话不多说了,直接上代码吧!

import tensorflow as tf
from PIL import Image
import matplotlib.pyplot as plt
import numpy as np
import os
i = 0
j = 0
num_shards = 100#总共写入的文件个数
instances_per_shard = 2#每个文件中的数据个数
sess=tf.InteractiveSession()
cwd = "F:/寒假/google--data/新建文件夹/" #图片数据所在目录位置(读者自己去改就好了)
classes = {'daisy','rose'} #预先自己定义的类别,根据自己的需要修改


def _int64_feature(value):#生成整数型的属性
   return tf.train.Feature(int64_list=tf.train.Int64List(value=[value]))

def _bytes_feature(value):#生成字符串型的属性
   return tf.train.Feature(bytes_list=tf.train.BytesList(value=[value]))
   
for index, name in enumerate(classes):#枚举函数
  class_path = cwd + name + "/"#选取具体数据目录
  for img_name in os.listdir(class_path):#遍历文件列表
    img_path = class_path + img_name#图片路径
    img = Image.open(img_path)
    img = img.resize((299, 299)) #图像reshape大小设置,根据自己的需要修改
    img_raw = img.tobytes()
      
    example = tf.train.Example(features=tf.train.Features(feature={
          'label': _int64_feature(index),
          'img_raw': _bytes_feature(img_raw),
          'i': _int64_feature(i),
          'j': _int64_feature(j)
        }))
    filename = ("F:/寒假/google--data/data.tfrecords-%.5d-of-%.5d"%(i,num_shards))
    if j == instances_per_shard-1:
      i+=1
    j+=1
    if j == instances_per_shard:
      j=0
    writer = tf.python_io.TFRecordWriter(filename)
      
    writer.write(example.SerializeToString())#将一个example写入tfrecord文件
writer.close()

以上这篇tensorflow生成多个tfrecord文件实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python入门之语句(if语句、while语句、for语句)
Jan 19 Python
python生成随机mac地址的方法
Mar 16 Python
tensorflow 加载部分变量的实例讲解
Jul 27 Python
dataframe 按条件替换某一列中的值方法
Jan 29 Python
python三引号输出方法
Feb 27 Python
Python可变和不可变、类的私有属性实例分析
May 31 Python
Python第三方库的几种安装方式(小结)
Apr 03 Python
Python爬虫实现百度翻译功能过程详解
May 29 Python
Python StringIO及BytesIO包使用方法解析
Jun 15 Python
解决keras GAN训练是loss不发生变化,accuracy一直为0.5的问题
Jul 02 Python
Python中过滤字符串列表的方法
Dec 22 Python
python的列表生成式,生成器和generator对象你了解吗
Mar 16 Python
tensorflow将图片保存为tfrecord和tfrecord的读取方式
Feb 17 #Python
Python 读取有公式cell的结果内容实例方法
Feb 17 #Python
Python破解BiliBili滑块验证码的思路详解(完美避开人机识别)
Feb 17 #Python
Tensorflow 实现将图像与标签数据转化为tfRecord文件
Feb 17 #Python
将自己的数据集制作成TFRecord格式教程
Feb 17 #Python
tensorflow 实现数据类型转换
Feb 17 #Python
Django Haystack 全文检索与关键词高亮的实现
Feb 17 #Python
You might like
WampServer下安装多个版本的PHP、mysql、apache图文教程
2015/01/07 PHP
PHP的Socket通信之UDP通信实例
2015/07/02 PHP
PHP 中TP5 Request 请求对象的实例详解
2017/07/31 PHP
利用php操作memcache缓存的基础方法示例
2017/08/02 PHP
jquery tools 系列 scrollable学习
2009/09/06 Javascript
ModelDialog JavaScript模态对话框类代码
2011/04/17 Javascript
document.documentElement和document.body区别介绍
2013/09/16 Javascript
用js提交表单解决一个页面有多个提交按钮的问题
2014/09/01 Javascript
Vue.js基础知识汇总
2016/04/27 Javascript
JavaScript中点击事件的写法
2016/06/28 Javascript
前端框架Vue.js中Directive知识详解
2016/09/12 Javascript
原生js轮播(仿慕课网)
2017/02/15 Javascript
Node.js创建HTTP文件服务器的使用示例
2018/05/11 Javascript
微信小程序使用map组件实现获取定位城市天气或者指定城市天气数据功能
2019/01/22 Javascript
Vue编程式跳转的实例代码详解
2019/07/10 Javascript
详解Vscode中使用Eslint终极配置大全
2019/11/08 Javascript
vue中渲染对象中属性时显示未定义的解决
2020/07/31 Javascript
[01:22:28]DOTA2-DPC中国联赛 正赛 SAG vs RNG BO3 第一场 1月18日
2021/03/11 DOTA
在Python的web框架中配置app的教程
2015/04/30 Python
python数组复制拷贝的实现方法
2015/06/09 Python
Python selenium抓取微博内容的示例代码
2018/05/17 Python
Python @property使用方法解析
2019/09/17 Python
Python的缺点和劣势分析
2019/11/19 Python
python绘制BA无标度网络示例代码
2019/11/21 Python
Windows下实现将Pascal VOC转化为TFRecords
2020/02/17 Python
Python 炫技操作之合并字典的七种方法
2020/04/10 Python
特色冷饮店创业计划书
2014/01/28 职场文书
倡议书格式
2014/04/14 职场文书
保密工作承诺书
2014/08/29 职场文书
留学推荐信中文范文
2015/03/26 职场文书
信用卡催款律师函
2015/05/27 职场文书
车间班组长竞聘书
2015/09/15 职场文书
2016教师年度考核评语大全
2015/12/01 职场文书
高一作文之暖冬
2019/11/09 职场文书
MySQL中distinct与group by之间的性能进行比较
2021/05/26 MySQL
Sleuth+logback 设置traceid 及自定义信息方式
2021/07/26 Java/Android