Commit 58c69c56 authored by Amelie Royer's avatar Amelie Royer

remove save_image_in_records for CIFAR

parent fa33c40f
......@@ -43,7 +43,7 @@ class CIFAR10Converter(Converter):
# Labels
self.label_names = unpickle(os.path.join(self.data_dir, 'batches.meta'))[b'label_names']
def convert(self, tfrecords_path, save_image_in_records=False):
def convert(self, tfrecords_path):
"""Convert the dataset in TFRecords saved in the given `tfrecords_path`"""
for name, data in self.data:
writer_path = '%s_%s' % (tfrecords_path, name)
......@@ -81,7 +81,7 @@ class CIFAR100Converter(CIFAR10Converter):
# Labels
self.label_names = unpickle(os.path.join(self.data_dir, 'meta'))[b'coarse_label_names']
def convert(self, tfrecords_path, save_image_in_records=False):
def convert(self, tfrecords_path):
"""Convert the dataset in TFRecords saved in the given `tfrecords_path`"""
for name, data in self.data:
writer_path = '%s_%s' % (tfrecords_path, name)
......
......@@ -1454,7 +1454,7 @@
"source": [
"%%time\n",
"converter = cifar.CIFAR10Converter(source_path)\n",
"converter.convert(target_path, save_image_in_records=save_image_in_records)"
"converter.convert(target_path)"
]
},
{
......@@ -1562,7 +1562,7 @@
"source": [
"%%time\n",
"converter = cifar.CIFAR100Converter(source_path)\n",
"converter.convert(target_path, save_image_in_records=save_image_in_records)"
"converter.convert(target_path)"
]
},
{
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