2017-09-12 10 views
1

この小さなプログラムで何がうまくいかないのか分かりません。ここでTensorflowプログラムがtfrecordを読んでいませんでした

はMCVEライターの抜粋です:

def convert_to_example(): 
    example = tf.train.Example(features=tf.train.Features(feature={ 
      'bboxes': _floats_feature([0.,1.]) 
      })) 
    return example 

writer = tf.python_io.TFRecordWriter(output_file) 
... 
for filename in filenames: 
    ... 
    example = convert_to_example() 
    writer.write(example.SerializeToString()) 

writer.close() 

これは私が例を読む方法です:

filename = '/path/to/file' 
record_iter = tf.python_io.tf_record_iterator(path=filename) 
example = tf.train.Example() 
l = [] 
for record in record_iter: 
    example.ParseFromString(record) 
    bboxes = example.features.feature['bboxes'].float_list.value[:] 
    l.append(bboxes) 

print(l) 

私は問題を絞り込んています

  • それはbytes_list
  • と一緒に動作しますで動作しますリストは一つだけ、整数ではなく、整数のリストである場合
  • 私はfloatのリストを使用する場合は、リストのfloatのリストは、

だから1つのフロートですがない場合、それはfloat_listで動作します/実行がデッドロックまたはクラッシュに達した場合。フロート/整数を使用すると、すべてがスムーズに実行されます。

答えて

0

このエラーはシステムによって異なります。ワークステーションでうまく動作しますが、私のPCではありません。 github.comに問題を公開しました。

関連する問題