Skip to content

Repository files navigation

MNIST 手写数字识别

启动 Web 页面

项目不需要额外安装 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。

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

Packages

Contributors

Languages