文章目录
- 一、前期工作
-
- 1. 设置GPU
- 2. 导入数据
- 3. 查看数据
- 二、数据预处理
-
- 1. 加载数据
- 2. 可视化数据
- 3. 再次检查数据
- 4. 配置数据集
- 三、AlexNet介绍
- 四、构建AlexNet网络模型
- 五、编译
- 六、训练模型
- 七、模型评估
- 八、保存and加载模型
- 九、预测
一、前期工作
本文将采用AlexNet
实现鸟类图片的识别分类
我的环境:
- 语言环境:Python3.6.5
- 编译器:jupyter notebook
- 深度学习环境:TensorFlow2.4.1
- 数据和代码: