本文共5部分，内容结构如下：1. 数据预处理：划分数据集。2. 加载自定义数据集。3. 建立Alexnet模型。4. 模型训练。5. 模型的评估和使用训练好的模型进行图片分类。6. 项目注意事项与测试结果展示。该项目可顺序复制上述代码到与文件名相同的文件中，组织相同的目录结构，配置好相应的pytorch环境，就可以顺利运行。本文所写的代码，都有较完整的注释，供大家研究以理解代码。

本文将分享自己写的3个函数，分别用来实现保存数据集图片的路径到txt文件，读取txt文件，划分训练集与验证集并保存到txt文件。