先安装环境 ----> 使用data_classify.py文件进行训练集与测试集分割 ----> 在进行训练即可
数据准备:当前数据存放 data_name 文件夹内
文件夹名就是类别名,n个类别就是n个文件夹
目录主要结构组成:
model_AlexNet.py ----> 自己建的AlexNet模型(可选其他模型)
model_Vgg16.py ----> pytorch自带更改的模型(可选其他模型)
train.py ----> 用于训练模型
test.py ----> 用于测试模型
辅助文件:
data_classify.py ----> 将 data_name内的类别分为训练集与测试集。
注意查看代码内容,包含argparse模块
清除单通道图像 -----> 数据清洗,处理异常图像
旧版数据加载 -----> 用于学习图像 数据加载
1