# MNIST 数据集

使用 PyTorch 训练

## 依赖安装

首先根据 [PyTorch 官网](https://pytorch.org/get-started/locally/) 安装 PyTorch，以便启用 GPU 加速

然后安装其他依赖

```shell
pip install fastapi uvicorn pillow tqdm
```

## 训练模型

首先你当然要先训练模型啦

```shell
python mnist_train.py --save-model
```

## 使用方法

训练完模型后，本仓库包含两种使用方法，即本地文件和 Web UI

### 本地文件

将图片文件放到 input 文件夹内，必须是 28*28 的灰度png格式，然后执行

```shell
python mnist.test.py
```

### Web UI

执行以下命令启动 Web UI

```shell
uvicorn webui:app
```

然后打开 <http://127.0.0.1:8000> 即可看到网页