2016-12-23 24 views

回答

0

Tensorflow不尋找單個tfrecord文件。因此,請隨時將您的「數據目錄」和「培訓目錄」指向已設置tfrecord文件的位置。

另外,請記住文件應該在各自的目錄中,例如TRAIN - *。tfrecord文件中的「train directory」。

如果您提到您要在這些TF記錄文件上運行的目標TF的型號,那麼可以更具體地回答問題。

希望它有幫助。

+0

你能解釋一下如何加載我的數據集,即我的'train_directory'我有5個標籤文件夾名稱爲'label_i'(i = {1,2,3,4,5})。 –

+0

^在每個'label_i'文件夾中,我都有單獨的TF記錄文件的圖像。 –

+0

錯了! Donot爲每個「標籤」/「類」文件夾創建單獨的TFRecord文件。在生成TFrecord文件時,指向父級data_dir,該父級data_dir中包含標籤目錄,並在build_dataset腳本中指定要在培訓中使用的圖像數量以及評估數量 – Anuj

相關問題