VGG16+CIFAR10分类的实现

import numpy as np
import tensorflow as tf
import os
import pickle
from keras.utils import np_utils

CIFAR_PATH = r’D:\python\MNIST_data\cifar-10-batches-py’

def load_data(file):
with open(file, ‘rb’) as f:
data = pickle.load(f, encoding=‘bytes’)
return data[b’data’], data[b’labels’]

def Cifar(filenames):
all_data = []
all_label = []
for file_name in filenames:
data, label = load_data(file_name)
all_data.append(data)
all_label.append(label)
all_data = np.vstack(all_data).astype(np.float32)
all_label = np.hstack(all_label)
#上面两行只是为了获取numpy类型的数组也可以向下面这么写：
/*
all_data = np.array(all_data).reshape([-1, 3072]).astype(np.float32)
all_label = np.array(all_label).reshape([len(all_data), -1])
*/
all_data /= 255
return all_data, all_label

通过索引打乱训练集

/*