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