# LeNet5

**Repository Path**: huiyideyi_admin/le-net5

## Basic Information

- **Project Name**: LeNet5
- **Description**: 图像识别神经网络LeNet5，可根据不同数据集训练出特定功能的模型.
- **Primary Language**: Python
- **License**: MIT
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 2
- **Forks**: 0
- **Created**: 2025-06-09
- **Last Updated**: 2025-10-07

## Categories & Tags

**Categories**: Uncategorized

**Tags**: 机器学习, 神经网络

## README

# LeNet系列神经网络模型

## 介绍
### CNN卷积神经网络，图像识别LeNet5，可根据不同数据集训练出特定功能的模型。
### 基于当前数据集的模型准确率在87.8%左右
![img_1.png](pics/img_1.png)
![img_2.png](pics/img_2.png)
![img_4.png](pics/img_4.png)
#### 软件架构
五层架构，两层卷积层，三层池化层，三层全连接层。
共61706个参数，目前训练准确率在87.9%左右，内置已经训练好的模型参数

### 安装环境，推荐使用Anaconda安装
1. Python==3.8
2. Pytorch==1.10.1
4. Cudatookit==11.3
5. Cudnn==8.2
6. torchsummary==1.5.1
7. numpy==1.23.2
8. pandas==1.3.4
9. matplotlib==3.5.0
10. sklearn==0.0

### 安装教程

1. 下载Anaconda [https://anaconda.org/](https://anaconda.org/)
2. 安装Anaconda
    ###### 创建仓库 conda create -n yourname python==3.8
    ###### 激活仓库 activate yourname
    ### 安装CPU版本Pytorch
    ###### 安装torch-cpu pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html 
    ### 安装GPU版本
    ###### 先检查本地GPU cuda版本 命令：nvidia-smi。然后根据cuda版本安装pytorch，cuda版本是向下兼容的，若为找到硬件的cuda版本，可选择低版本向下兼容conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge
    ### 安装其他库
    ###### pip install torchsummary==1.5.1
    ###### pip install numpy==1.23.2
    ###### pip install pandas==1.3.4
    ###### pip install matplotlib==3.5.0
    ###### pip install sklearn==0.0
### 注意事项
![img_3.png](pics/img_3.png)
###### 若出现上述错误，解决方案：conda list找到torchvision 其中的版本号0.11.2，手动修改为0.11.0。conda将重新下载匹配所需的系列环境即可解决

#### 参与贡献

1.  yKun