项目不需要额外安装 Web 框架,直接使用现有虚拟环境启动:
minst/bin/python app.py然后访问 http://127.0.0.1:8000。
服务启动时会同时加载以下两个模型,可直接在网页上切换:
- 全连接模型:
digit_classifier.pth - CNN 模型:
conv_classifier.pth
如需替换全连接模型或 CNN 模型的权重文件:
minst/bin/python app.py \
--model digit_classifier_gpu.pth \
--conv-model conv_classifier.pth服务端始终使用 CPU 推理,因此不要求浏览器端或运行环境具有 GPU。