Converts cifar-10 numpy lists to python lists

cifar10_converter

import pickle


def unpickle(file):
    with open(file, 'rb') as fo:
        dic = pickle.load(fo, encoding='bytes')
        fo.close()
    return dic


def repickle(file, data):
    with open(file, 'wb') as f:
        pickle.dump(data, f)


def convert_cifar10(from_folder, to_folder):
    test_data = unpickle('%s/test_batch' % from_folder)
    test_data[b'data'] = test_data[b'data'].tolist()
    repickle('%s/test_batch' % to_folder, test_data)

    for i in range(1, 6):
        training_data = unpickle('%s/data_batch_%d' % (from_folder, i))
        training_data[b'data'] = training_data[b'data'].tolist()
        repickle('%s/data_batch_%d' % (to_folder, i), training_data)


# Make sure you have the './cifar10_list' created first
convert_cifar10('./cifar10', './cifar10_list')