MNIST 数据集

使用 PyTorch 训练

依赖安装

首先根据 PyTorch 官网安装 PyTorch，以便启用 GPU 加速

然后安装其他依赖

pip install fastapi uvicorn pillow tqdm

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

python mnist_train.py --save-model

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

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

python mnist.test.py

执行以下命令启动 Web UI

uvicorn webui:app

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