深度学习中的CNN模型用于图像分类任务(基于cifar10数据集)
发布时间
阅读量:
阅读量
Alexnet模型在CIFAR10分类中的应用
import tensorflow as tf
import numpy as np
import pickle
import os
# 读取文件目录
# 自行下载数据集
cifar_dir = './cifar-10-batches-py'
print(os.listdir(cifar_dir))
train_filenames = [os.path.join(cifar_dir,'data_batch_%d'%i) for i in range(1,6)]
test_filenames = [os.path.join(cifar_dir,'test_batch') ]
# 构建数据载入函数
def load_data(filename):
with open(filename,'rb') as f:
data = pickle.load(f,encoding='bytes')
return data[b'data'],data[b'labels']
# 构建数据处理类
# 主要完成数据的提取,归一化,
全部评论 (0)
还没有任何评论哟~
