以CIFAR10数据为例的分类器
实验课作业,由于是很经典的分类任务,所以整理了一下记录下来,实际上TensorFlow源码中就有很好的CIFAR10示例(包含单机和多样化版本),不过既然要交作业,自己的在基础CNN分类的版本外,添加了使用ResNet进行分类的实验,收敛速度远快于基础CNN。
一,文件介绍
公用脚本
ops.py网络层封装实现,已被Advanced_CNN.py和ResNet.py调用cifar10_input.py :数据读入相关函数脚本,包含对训练数据和测试数据的不同预先路径设置eval_CNN.py :测试用eval_CNN.py ,读取./images目录下的图片文件,并输出对应的预测结果
基础CNN分类器相关脚本
Advanced_CNN.py :使用CNN的分类器,脚本本身包含了网络构建和训练相关的摘要,可以直接运行训练数据
ResNet相关脚本
ResNet
2021-11-20 15:45:01
979KB
系统开源
1