自製資料集轉為tfrecord格式

2021-10-01 02:53:36 字數 730 閱讀 9200

教程

上面這個教程中轉化格式那部分很好用,親測有效,python程式改為自己的路徑執行的時候,因為是第一次轉化這種格式,也沒有怎麼用過遠端伺服器,所以在csv轉tfrecord的時候,一直出現這個錯誤:

網上也找不到,最後問師兄的時候,突然發現是疏忽了下面這兩行**是需要自己輸入路徑的,導致卡了一整天,這兒也不對,那兒也不對(哎,自己太蠢了):

flags.define_string(

'csv_input',''

,'path to the csv input'

)flags.define_string(

'output_path',''

,'path to output tfrecord'

)

具體的輸入方法,舉個例子:

python3 generate_tfrecord.py --csv_input /home/tensorflow/models/research/object_detection/images/train/meter_train.csv --output_path /home/tensorflow/models/research/object_detection/images/train/meter_train.tfrecord

貓狗大戰的TFrecord資料集製作

import tensorflow as tf import numpy as np import os from pil import image 沒有下面兩句德華會出現 image file is truncate的問題 from pil import imagefile imagefile.l...

tf如何自製使用資料集

這裡用tf.data.dataset.from tensor slices進行載入資料集 第一步,準備numpy資料 def load mnist batch size,is training true if is training fd open r data mnist train images...

《2020 02 12》tfrecord讀寫資料

僅供自己學習使用 tensorflow通過tfrecord高效讀寫資料 利用tensorflow提供的tfrecord資料儲存格式工具,我們可以將我們已經進行過處理的資料儲存起來,以便我們下次更高效地讀取,略過資料處理的過程,提高效率。具體的步驟大概分為以下幾步 將資料轉化為tf.train.fea...