Advertisement

深度学习中的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)

还没有任何评论哟~