图像分类项目cifar10-tensorflow 代码解读+视频讲解

一、前言

代码来源于 cifar10-tensorflow  ，代码作者的帖子 如何一步一步提高图像分类准确率？

本系列目的：从头开始学习图像分类，作为深度学习的一个基础项目，用视频+博客的方式记录代码细节，分享给大家，希望有所帮助。

我录的视频：https://www.bilibili.com/video/BV1Cf4y1k7es/

二、准备

tensorflow安装

我以前用过tf1，本项目也是基于tf1版本

我输入nvidia-smi后，得到cuda版本为10.1  ，python版本3.6，这里我安装tf1.15

conda install --channel https://conda.anaconda.org/hanyucui tensorflow-gpu=1.15

安装细节可参考https://zhuanlan.zhihu.com/p/109067934

检测是否安装tensorflow-gpu成功，输入以下命令，如果为True则成功。

python -c "import tensorflow as tf;print(tf.test.is_gpu_available())"

数据准备

下载地址：http://www.cs.toronto.edu/~kriz/cifar.html   选择以下版本： 

CIFAR-10 python version

163 MB

c58f30108f718f92721af3b95e74349a

 下载慢可参考：https://zhuanlan.zhihu.com/p/129078357

三、项目启动

 

四、数据模块

 

五、网络模块

 

六、训练管理