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 urlopen 使用小示例
Sep 06 Python
使用70行Python代码实现一个递归下降解析器的教程
Apr 17 Python
最基础的Python的socket编程入门教程
Apr 23 Python
星球大战与Python之间的那些事
Jan 07 Python
python特性语法之遍历、公共方法、引用
Aug 08 Python
Python异常处理知识点总结
Feb 18 Python
在SQLite-Python中实现返回、查询中文字段的方法
Jul 17 Python
pygame编写音乐播放器的实现代码示例
Nov 19 Python
tensorflow求导和梯度计算实例
Jan 23 Python
在django admin详情表单显示中添加自定义控件的实现
Mar 11 Python
Python的历史与优缺点整理
May 26 Python
python与c语言的语法有哪些不一样的
Sep 13 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
Zend公司全球首推PHP认证
2006/10/09 PHP
php采集文章中的图片获取替换到本地(实现代码)
2013/07/08 PHP
PHP实现页面静态化的超简单方法
2016/09/06 PHP
PHP中include()与require()的区别说明
2017/02/14 PHP
js 编写规范
2010/03/03 Javascript
jQuery插件开发基础简单介绍
2013/01/07 Javascript
利用jQuery实现漂亮的圆形进度条倒计时插件
2015/09/30 Javascript
使用Promise解决多层异步调用的简单学习心得
2016/05/17 Javascript
jQuery实现查找链接文字替换属性的方法
2016/06/27 Javascript
浅谈jQuery中hide和fadeOut的区别 show和fadeIn的区别
2016/08/18 Javascript
基于JavaScript实现跳转提示页面
2016/09/24 Javascript
JS实现的相册图片左右滚动完整实例
2016/11/23 Javascript
前端 Vue.js 和 MVVM 详细介绍
2016/12/29 Javascript
Angular2开发——组件规划篇
2017/03/28 Javascript
vue路由对不同界面进行传参及跳转的总结
2019/04/20 Javascript
深入理解javascript prototype的相关知识
2019/09/19 Javascript
layui实现给某一列加点击事件
2019/10/26 Javascript
Vue extend的基本用法(实例详解)
2019/12/09 Javascript
解决三元运算符 报错“SyntaxError: can''t assign to conditional expression”
2020/02/12 Javascript
Python程序员开发中常犯的10个错误
2014/07/07 Python
python一键升级所有pip package的方法
2017/01/16 Python
python实现飞机大战微信小游戏
2020/03/21 Python
Python3列表内置方法大全及示例代码小结
2019/05/10 Python
Python环境下安装PyGame和PyOpenGL的方法
2020/03/25 Python
Python datetime模块的使用示例
2021/02/02 Python
CSS3截取字符串实例代码【推荐】
2018/06/07 HTML / CSS
CSS3的calc()做响应模式布局的实现方法
2017/09/06 HTML / CSS
Python里面search()和match()的区别
2016/09/21 面试题
给老婆的搞笑检讨书
2014/01/12 职场文书
工作评语大全
2014/04/26 职场文书
导师就业推荐信范文
2014/05/22 职场文书
自动化专业毕业生求职信
2014/06/18 职场文书
群众路线剖析材料
2014/09/30 职场文书
北京天坛导游词
2015/02/12 职场文书
大学生自我评价范文
2015/03/03 职场文书
烈士陵园扫墓感想
2015/08/07 职场文书