CIFAR10 BQ

cifar10_bq.py

import tensorflow as tf
import tensorflow_datasets as tfds

ds = tfds.load('cifar10', as_supervised=True)

def csvfy(dataset, filename):
  with open(filename, 'w') as f:
    for image, label in dataset:
      f.write('{},'.format(label.numpy()))
      f.write(','.join(str(x) for x in list(image.numpy().reshape(-1))))
      f.write('\n')

csvfy(ds['train'], 'train.csv')
csvfy(ds['test'], 'test.csv')