目录
1. 介绍
之前做的用PyQT做的GUI窗口分类图像,然后打包成exe的方法有些过时了
这次用的方法时用streamlit将神经网络部署成一个web
项目包含的文件:
- LeNet.pth 是训练好的网络参数
- predict 是做预测用的
- main 是搭建web 的代码
- image 包含一个dog.jpg 是默认web显示的图片
- 剩下两个txt文件这里可以不需要
参考的是这篇文章:pytorch 实现图像分类+web部署
网站的链接为:图像分类
github 的链接为: