猫狗分类学习项目

说明文档：本项目用于学习图像分类的教学项目

数据集下载：

链接: https://pan.baidu.com/s/1vK2ITIjAPiZUQD7OfsEc8A 密码: 1k51

先安装环境 ----> 使用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模块

清除单通道图像 -----> 数据清洗，处理异常图像

旧版数据加载 -----> 用于学习图像 数据加载

安装环境：

pip install -r requirements.txt

pip install tensorboard

去pytorch官方下载：自己对应cuda版本的pytorch，如果没有GPU 下载cpu也可

下面是我的下载：

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

报错：

UserWarning: Detected call of lr_scheduler.step() before optimizer.step(). In PyTorch 1.1.0 and later, you should call them in the opposite order: optimizer.step() before lr_scheduler.step(). Failure to do this will result in PyTorch skipping the first value of the learning rate schedule. See more details at https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate warnings.warn("Detected call of lr_scheduler.step() before optimizer.step(). "

原因是数据有些问题，但不影响训练，使用者可更换自己的数据集

完整代码下载地址：https://download.csdn.net/download/shiyunzhe2021/87522210