def _convert_data(split): def _create_example(item): image, label = item example = tf.train.Example(features=tf.train.Features( feature={ 'image': tf.train.Feature( bytes_list=tf.train.BytesList(value=[image.tobytes()])), 'label': tf.train.Feature( int64_list=tf.train.Int64List(value=[label.astype(np.int64)])) })) return example utils.parallel_record_writer( _image_iterator(split), _create_example, get_split(split))
def _convert_data(split): """Convert the dataset to TFRecord format.""" def _create_example(item): image, label = item example = tf.train.Example(features=tf.train.Features( feature={ 'image': tf.train.Feature( bytes_list=tf.train.BytesList(value=[image])), 'label': tf.train.Feature( int64_list=tf.train.Int64List(value=[label.astype(np.int64)])) })) return example utils.parallel_record_writer( _image_iterator(split), _create_example, get_split(split))