44 lines
738 B
Markdown
44 lines
738 B
Markdown
|
# 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> 即可看到网页
|