猫狗分类学习项目.md · 李博文/图像分类-猫狗

## 猫狗分类学习项目

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

数据集下载：

链接: 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()`. "

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

李博文/图像分类-猫狗

简介

发行版

贡献者

近期动态

李博文/图像分类-猫狗 .gitee-modal { width: 500px !important; }

简介

发行版

贡献者

近期动态

搜索帮助

李博文/图像分类-猫狗