CNN-Learning

基于 PyTorch 实现经典卷积神经网络模型，包括：

• LeNet
• AlexNet
• VGG
• ResNet

用于学习和实践卷积神经网络的基本结构与训练流程。

项目结构

CNN-Learning/
│
├── AlexNet/
├── LeNet/
├── ResNet/
├── VGG/
│
├── requirements.txt
├── README.md
└── .gitignore

环境配置

安装依赖：

pip install -r requirements.txt

数据集

本项目使用 FashionMNIST 数据集。

数据集会在首次运行时自动下载。

模型训练

以 LeNet 为例：

cd LeNet
python model_train.py

模型测试

python model_test.py

已实现模型

模型	状态
LeNet	✅
AlexNet	✅
VGG	✅
ResNet	✅

开发环境

• Python 3.x
• PyTorch
• torchvision
• NumPy
• Matplotlib

说明

本项目主要用于学习经典 CNN 网络结构及其实现方法。