快捷導(dǎo)航

tensorflow生成多個(gè)tfrecord文件實(shí)例

更新時(shí)間：2020年02月17日 10:07:02 作者：zhx_123987

今天小編就為大家分享一篇tensorflow生成多個(gè)tfrecord文件實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

我就廢話不多說(shuō)了，直接上代碼吧！

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#總共寫入的文件個(gè)數(shù)
instances_per_shard = 2#每個(gè)文件中的數(shù)據(jù)個(gè)數(shù)
sess=tf.InteractiveSession()
cwd = "F:/寒假/google--data/新建文件夾/" #圖片數(shù)據(jù)所在目錄位置(讀者自己去改就好了)
classes = {'daisy','rose'} #預(yù)先自己定義的類別，根據(jù)自己的需要修改


def _int64_feature(value):#生成整數(shù)型的屬性
   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):#枚舉函數(shù)
  class_path = cwd + name + "/"#選取具體數(shù)據(jù)目錄
  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大小設(shè)置，根據(jù)自己的需要修改
    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())#將一個(gè)example寫入tfrecord文件
writer.close()

以上這篇tensorflow生成多個(gè)tfrecord文件實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Django使用mysqlclient服務(wù)連接并寫入數(shù)據(jù)庫(kù)的操作過(guò)程
這篇文章主要介紹了Django使用mysqlclient服務(wù)連接并寫入數(shù)據(jù)庫(kù),本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下
2022-07-07
python將字符串轉(zhuǎn)換成數(shù)組的方法
這篇文章主要介紹了python將字符串轉(zhuǎn)換成數(shù)組的方法,涉及Python操作字符串與數(shù)組的相關(guān)技巧,非常具有實(shí)用價(jià)值,需要的朋友可以參考下
2015-04-04
Python淘寶秒殺的腳本實(shí)現(xiàn)
這篇文章主要介紹了Python淘寶秒殺的腳本實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2023-01-01
Django框架下靜態(tài)模板的繼承操作示例
這篇文章主要介紹了Django框架下靜態(tài)模板的繼承操作,結(jié)合實(shí)例形式分析了Django框架模板繼承操作的相關(guān)原理與操作注意事項(xiàng),需要的朋友可以參考下
2019-11-11
Python實(shí)現(xiàn)在線程里運(yùn)行scrapy的方法
這篇文章主要介紹了Python實(shí)現(xiàn)在線程里運(yùn)行scrapy的方法,涉及Python線程操作的技巧,非常具有實(shí)用價(jià)值,需要的朋友可以參考下
2015-04-04
Python如何建立多個(gè)值和單個(gè)鍵的映射
在Python中,常見(jiàn)的字典只能映射單個(gè)鍵到單個(gè)值,若需映射單個(gè)鍵到多值,可以通過(guò)將值存儲(chǔ)于列表或集合中實(shí)現(xiàn),使用列表可以保持元素插入順序,而使用集合則可以去重,collections模塊的defaultdict類簡(jiǎn)化了此類多值字典的創(chuàng)建過(guò)程
2024-09-09
Python+unittest+requests+excel實(shí)現(xiàn)接口自動(dòng)化測(cè)試框架
這篇文章主要介紹了Python+unittest+requests+excel實(shí)現(xiàn)接口自動(dòng)化測(cè)試框架，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2020-12-12
使用pytorch完成kaggle貓狗圖像識(shí)別方式
今天小編就為大家分享一篇使用pytorch完成kaggle貓狗圖像識(shí)別方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2020-01-01
python中數(shù)組和列表的簡(jiǎn)單實(shí)例
,Python數(shù)據(jù)類型分為很多種,其中包括元組、字典、列表等,下面這篇文章主要給大家介紹了python中數(shù)組和列表使用的相關(guān)資料,以及python列表和Numpy數(shù)組的區(qū)別,需要的朋友可以參考下
2022-03-03
Pandas查詢數(shù)據(jù)df.query的使用
本文主要介紹了Pandas查詢數(shù)據(jù)df.query的使用，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2022-07-07