TensorFlowのtfrecordファイル内のデータ数を数える
ちょっと感心したのでメモ。
TensorFlowで自作のデータセット(tfrecord)を作成したはいいけど、正確なデータ数を数えるの忘れてたってこと、あるよねえ。ねえか。
いざ、自力でやろうとすると、、、あれ?どうすんの?となった。
学習時のコードを見ると、データの中から指定した数だけ持ってきて学習に使う、という処理なので実は正確な数がわからなくても問題ない。どうするんだろうと思って調べたら、、、あるやん!
Obtaining total number of records from .tfrecords file in Tensorflow - Stack Overflow
↑おかげで一撃! tf.python_io.tf_record_iterator(fn) なんて便利なのがあったのね!!
TensorFlowで自作のデータセット(tfrecord)を作成したはいいけど、正確なデータ数を数えるの忘れてたってこと、あるよねえ。ねえか。
いざ、自力でやろうとすると、、、あれ?どうすんの?となった。
学習時のコードを見ると、データの中から指定した数だけ持ってきて学習に使う、という処理なので実は正確な数がわからなくても問題ない。どうするんだろうと思って調べたら、、、あるやん!
Obtaining total number of records from .tfrecords file in Tensorflow - Stack Overflow
↑おかげで一撃! tf.python_io.tf_record_iterator(fn) なんて便利なのがあったのね!!
スポンサーサイト