tfrecords資料的讀寫

2021-08-20 19:54:59 字數 1851 閱讀 7187

tfrecords_write.py

import tensorflow as tf

from tensorflow.examples.tutorials.mnist import input_data

import numpy as np

#定義乙個writer->轉化資料的格式->包裝好乙個example->將example序列化->寫入

#將資料轉化成相應的格式

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]))

mnist = input_data.read_data_sets('../mnist_data',one_hot=true,dtype=tf.uint8)

images = mnist.train.images

labels = mnist.train.labels

pixels = images.shape[1]

num_examples = mnist.train.num_examples

filename = 'output.tfrecords'

writer = tf.python_io.tfrecordwriter(filename)

for i in range(num_examples):

image_raw = images[i].tostring()

example = tf.train.example(features=tf.train.features(feature=))

writer.write(example.serializetostring())

writer.close()

tfrecords_read.py

import tensorflow as tf

#定義reader->read乙個example->解析->列印

reader = tf.tfrecordreader()

#使用佇列來維護輸入檔案列表

filename_queue = tf.train.string_input_producer(['output.tfrecords'])

#獲取序列化的資料,每次只讀取乙個

_,serialize_examples = reader.read(filename_queue)

#解析資料

features = tf.parse_single_example(

serialize_examples,

features=

)#將字串解析成畫素陣列

image = tf.decode_raw(features['image_raw'],tf.uint8)

label = tf.cast(features['label'],tf.int32)

pixels = tf.cast(features['pixels'],tf.int32)

with tf.session() as sess:

#啟動多執行緒

threads = tf.train.start_queue_runners(sess,tf.train.coordinator())

for i in range(10):

#每次都隨機讀取乙個樣例

print(sess.run([label]))

SSD訓練資料tfrecords資料產生bug

訓練資料tfrecords產生的問題 info tensorflow error reported to coordinator all bounding box coordinates must be in 0.0,1.0 1.002提供兩條解決思路 首先,第乙個方法是可行的,就是比較麻煩,要找到...

TFRecords 檔案的生成和讀取

1.tensorflow提供了tfrecords的格式來統一儲存資料,理論上,tfrecords可以儲存任何形式的資料。tfrecords檔案中的資料都是通過tf.train.example protocol buffer的格式儲存的。以下的 給出了tf.train.example的定義。messa...

R的資料讀寫

在使用任何一款資料分析軟體的時候,首先要做的就是資料成功的讀寫問題,所以不同於其他文件的書寫方法,本文將 如何讀寫本地文字檔案。作業系統 win10 r版本 r 3.2.4 win rstudio版本 rstudio 0.99.896 對於許多初學者而言,本身並沒有足夠的資料集合進行資料探索,在此推...