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翻译软件实现代码(使用google api完成)
Nov 26 Python
Python调用ctypes使用C函数printf的方法
Aug 23 Python
python3写爬取B站视频弹幕功能
Dec 22 Python
Python MySQLdb 使用utf-8 编码插入中文数据问题
Mar 13 Python
Python使用Django实现博客系统完整版
Sep 29 Python
Python查找第n个子串的技巧分享
Jun 27 Python
用Python分析3天破10亿的《我不是药神》到底神在哪?
Jul 12 Python
Python文件常见操作实例分析【读写、遍历】
Dec 10 Python
Appium Python自动化测试之环境搭建的步骤
Jan 23 Python
TensorFlow tf.nn.conv2d实现卷积的方式
Jan 03 Python
Python批量将图片灰度化的实现代码
Apr 11 Python
Python通过文本和图片生成词云图
May 21 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
ThinkPHP关联模型操作实例分析
2012/09/23 PHP
PHP上传图片进行等比缩放可增加水印功能
2014/01/13 PHP
php结合正则获取字符串中数字
2015/06/19 PHP
PHP+shell脚本操作Memcached和Apache Status的实例分享
2016/03/11 PHP
thinkphp项目部署到Linux服务器上报错“模板不存在”如何解决
2016/04/27 PHP
CodeIgniter框架常见用法工作总结
2017/03/16 PHP
javascript 写类方式之十
2009/07/05 Javascript
对象特征检测法判断浏览器对javascript对象的支持
2009/07/25 Javascript
javascript删除字符串最后一个字符
2014/01/14 Javascript
javascript中的nextSibling使用陷(da)阱(keng)
2014/05/05 Javascript
jQuery实现的兼容性浮动层示例
2016/08/02 Javascript
JS限定手机版中图片大小随分辨率自动调整的方法
2016/12/05 Javascript
React-Native实现ListView组件之上拉刷新实例(iOS和Android通用)
2017/07/11 Javascript
通过jquery获取上传文件名称、类型和大小的实现代码
2018/04/19 jQuery
详解Nodejs mongoose
2018/06/10 NodeJs
JS中的函数与对象的创建方式
2019/05/12 Javascript
vue实现购物车选择功能
2020/01/10 Javascript
原生js实现的观察者和订阅者模式简单示例
2020/04/18 Javascript
[47:43]Alliance vs KG 2019国际邀请赛小组赛 BO2 第一场 8.16
2019/08/18 DOTA
解决windows下Sublime Text 2 运行 PyQt 不显示的方法分享
2014/06/18 Python
Python map和reduce函数用法示例
2015/02/26 Python
Python中IPYTHON入门实例
2015/05/11 Python
Python3控制路由器——使用requests重启极路由.py
2016/05/11 Python
python基础教程之五种数据类型详解
2017/01/12 Python
Windows下的Python 3.6.1的下载与安装图文详解(适合32位和64位)
2018/02/21 Python
python range实例用法分享
2020/02/06 Python
html5 Canvas画图教程(2)—画直线与设置线条的样式如颜色/端点/交汇点
2013/01/09 HTML / CSS
Expedia印度:您的一站式在线旅游网站
2017/08/24 全球购物
美国最大的半成品净菜电商:Blue Apron(蓝围裙)
2018/04/27 全球购物
新西兰便宜隐形眼镜购买网站:QUICKLENS New Zealand
2019/03/02 全球购物
自我评价的正确写法
2013/09/19 职场文书
八年级生物教学反思
2014/01/22 职场文书
地震捐款倡议书
2014/08/29 职场文书
银行自荐信怎么写
2015/03/05 职场文书
乡镇司法所2015年度工作总结
2015/10/14 职场文书
向Spring IOC 容器动态注册bean实现方式
2022/07/15 Java/Android