頭と尻尾はくれてやる!

パソコンおやじのiPhoneアプリ・サイト作成・運営日記


TensorFlowのtfrecordファイル内のデータ数を数える

ちょっと感心したのでメモ。
TensorFlowで自作のデータセット(tfrecord)を作成したはいいけど、正確なデータ数を数えるの忘れてたってこと、あるよねえ。ねえか。

いざ、自力でやろうとすると、、、あれ?どうすんの?となった。
学習時のコードを見ると、データの中から指定した数だけ持ってきて学習に使う、という処理なので実は正確な数がわからなくても問題ない。どうするんだろうと思って調べたら、、、あるやん!

Obtaining total number of records from .tfrecords file in Tensorflow - Stack Overflow

↑おかげで一撃! tf.python_io.tf_record_iterator(fn) なんて便利なのがあったのね!!

<< 解答の範囲検討  TopPage  matplotlibでグラフを描こうとするとエラー >>

コメント


管理者にだけ表示を許可する
 

トラックバック

トラックバックURL
http://ringsbell.blog117.fc2.com/tb.php/1119-07b28205




Copyright ©頭と尻尾はくれてやる!. Powered by FC2 Blog. Template by eriraha.

FC2Ad