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实现的Kmeans++算法实例
Apr 26 Python
在Python的Flask框架中使用日期和时间的教程
Apr 21 Python
实例说明Python中比较运算符的使用
May 13 Python
Python中encode()方法的使用简介
May 18 Python
python3抓取中文网页的方法
Jul 28 Python
Python 描述符(Descriptor)入门
Nov 20 Python
详解python发送各类邮件的主要方法
Dec 22 Python
tensorflow实现softma识别MNIST
Mar 12 Python
Python中py文件引用另一个py文件变量的方法
Apr 29 Python
详细介绍pandas的DataFrame的append方法使用
Jul 31 Python
详解如何在PyCharm控制台中输出彩色文字和背景
Aug 17 Python
python pymysql库的常用操作
Oct 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
ThinkPHP CURD方法之limit方法详解
2014/06/18 PHP
ThinkPHP自动完成中使用函数与回调方法实例
2014/11/29 PHP
Yii实现自动加载类地图的方法
2015/04/01 PHP
WordPress中转义HTML与过滤链接的相关PHP函数使用解析
2015/12/22 PHP
PHP中FTP相关函数小结
2016/07/15 PHP
PHP实现的登录页面信息提示功能示例
2017/07/24 PHP
jquery BS,dialog控件自适应大小
2009/07/06 Javascript
获取css样式表内样式的js函数currentStyle(IE),defaultView(FF)
2011/02/14 Javascript
jquery对单选框,多选框,文本框等常见操作小结
2014/01/08 Javascript
js+CSS实现弹出居中背景半透明div层的方法
2015/02/26 Javascript
JavaScript获取网页支持表单字符集的方法
2015/04/02 Javascript
使用bootstrap实现多窗口和拖动效果
2016/09/22 Javascript
AngularJS路由切换实现方法分析
2017/03/17 Javascript
非常实用的vue导航钩子
2017/03/20 Javascript
基于 webpack2 实现的多入口项目脚手架详解
2017/06/26 Javascript
NodeJs form-data格式传输文件的方法
2017/12/13 NodeJs
Javascript中绑定click事件的四种方式介绍
2018/10/26 Javascript
Vue自定义属性实例分析
2019/02/23 Javascript
使用JS判断页面是首次被加载还是刷新
2019/05/26 Javascript
python爬虫之百度API调用方法
2017/06/11 Python
详解 Python 读写XML文件的实例
2017/08/02 Python
Python实现的十进制小数与二进制小数相互转换功能
2017/10/12 Python
python实现flappy bird小游戏
2018/12/24 Python
python实现信号时域统计特征提取代码
2020/02/26 Python
使用 CSS3 中@media 实现网页自适应的示例代码
2020/03/24 HTML / CSS
巴西男士个人护理产品商店:SHOP4MEN
2017/08/07 全球购物
GoDaddy英国:全球排名第一的域名注册商
2018/06/08 全球购物
经典c++面试题四
2015/05/14 面试题
团队队名口号大全
2014/06/06 职场文书
2014年党的群众路线教育实践活动整改措施(个人版)
2014/09/25 职场文书
2014年管理人员工作总结
2014/12/01 职场文书
征求意见函
2015/06/05 职场文书
Python 解决空列表.append() 输出为None的问题
2021/05/23 Python
深入浅析Django MTV模式
2021/09/04 Python
36个正则表达式(开发效率提高80%)
2021/11/17 Javascript
JavaScript中MutationObServer监听DOM元素详情
2021/11/27 Javascript