Commit d12b00da authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Updated the LFW to tfrecord script to the new feature format

parent 0c76788e
Pipeline #13582 failed with stages
in 23 minutes and 20 seconds
......@@ -95,8 +95,10 @@ def main(argv=None):
data = bob.io.image.to_matplotlib(bob.io.base.load(path)).astype(data_type)
data = data.tostring()
feature = {'train/data': _bytes_feature(data),
'train/label': _int64_feature(file_to_label(client_ids, f))}
feature = {'data': _bytes_feature(data),
'label': _int64_feature(file_to_label(client_ids, f)),
'key': _bytes_feature(str(f.path)),
}
example = tf.train.Example(features=tf.train.Features(feature=feature))
writer.write(example.SerializeToString())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment