Commit a6379bce authored by Maria Lande's avatar Maria Lande
Browse files

updated csv and record

parent a6878408
No preview for this file type
This diff is collapsed.
......@@ -35,6 +35,6 @@ def main():
'''
image_path = os.path.join(os.getcwd())
xml_df = xml_to_csv(image_path)
xml_df.to_csv('data/train_labels.csv', index=None)
xml_df.to_csv('../../../data/train_labels.csv', index=None)
main()
\ No newline at end of file
# reads one element from the train.record file
import tensorflow as tf
raw_dataset = tf.data.TFRecordDataset("data/train.record")
for raw_record in raw_dataset.take(1):
example = tf.train.Example()
example.ParseFromString(raw_record.numpy())
print(example)
\ No newline at end of file
Supports Markdown
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