文章目录 一、前期工作 1. 设置GPU 2. 导入数据 3. 查看数据 二、数据预处理 1. 加载数据 2. 可视化数据 3. 再次检查数据 4. 配置数据集 三、AlexNet介绍 四、构建AlexNet网络模型 五、编译 六、训练模型 七、模型评估 八、保存and加载模型 九、预测 一、前期工作 本文将采用AlexNet实现鸟类图片的识别分类 我的环境: 语言环境:Python3.6.5 编译器:jupyter notebook 深度学习环境:TensorFlow2.4.1 1. 设置GPU 如果使用的是CPU可以注释掉这部分的代码。 import tensorflow as tf gpus = tf.config.list_physical_devices("GPU") if gpus