Flask和Docker部署Pytorch深度学习模型
模型训练预先训练深度学习模型,保存训练后的模型或参数,此处采用图像分类模型。Flask搭建模型Flask是一个使用 Python 编写的轻量级Web应用框架,可通过$pip install flask进行安装。首先编写app.py文件import osfrom flask import Flask, render_template, request, redirectimport iofrom PIL import Imageimport uuidimport vision app
原创
2020-08-31 21:54:07 ·
3304 阅读 ·
5 评论