Python Flask基本使用以及与Pytorch整合

Flask为Python WEB框架，类似于Java的SpringMVC，Struts等，其作用是提供HTTP服务，可以通过浏览器访问并返回html（被浏览器解析为网页内容），json，xml等数据

一、引入相关模块及模块详解

引入flask相关模块：from flask import Flask, render_template,request

Flask:为web框架核心模块

request：接受HTTP请求的传参(GET，POST)

render_template：Flask框架的模板引擎（数据直接通过控制器映射到html模板中，类似于php的smarty框架），如果只返回json数据则无需引入

二.初始化Flask

app = Flask(__name__) #初始化Flask，并将对象赋值给app

if __name__ == '__main__':

app.run(host="0.0.0.0", port=80,debug = True) #当程序启动是运行run方法开启http服务器并绑定到80端口

三、定义一个路由向浏览器返回字符串，并测试浏览器传参

@app.route('/gainEntityByContext')

def gainEntityByContext():

print(request.args("name")) #客户端get传参

return 'Hello, World!'

四.返回json数据

@app.route("/upContent")

def upContent():

data = [ { 'a' : 1, 'b' : 2, 'c' : 3, 'd' : 4, 'e' : 5 } ]

return json.dumps(data)

五.加入html模板

@app.route("/testtemp")

def testTemp():

return render_template('test.htm', name="测试")

六.与pytorch结合，完成对深度学习模型的使用

@app.route("/loadlinear")

def loadLinear():

md = torch.load(basepath+request.args['mdname'])

x_data = np.linspace(-2, 2, 200)[:, np.newaxis]

x_data = x_data.reshape(-1,1)

x_data = torch.FloatTensor(x_data)

y_data = md(x_data)

plt.plot(x_data, y_data.data.numpy(), 'r-')

imgurl = 'static/'+request.args['mdname']+".png"

plt.savefig(imgurl)

return render_template('loadtest.htm', url=imgurl,name="测试模型加载")

七.全部代码

from flask import Flask, render_template,request
import torch
import json
import numpy as np
import matplotlib.pyplot as plt
app = Flask(__name__)
basepath = "modeldata/"
@app.route('/gainEntityByContext')
def gainEntityByContext():
print(request.args("name"))
return 'Hello, World!'
@app.route("/upContent")
def upContent():
data = [ { 'a' : 1, 'b' : 2, 'c' : 3, 'd' : 4, 'e' : 5 } ]
return json.dumps(data)
@app.route("/loadlinear")
def loadLinear():
md = torch.load(basepath+request.args['mdname'])
x_data = np.linspace(-2, 2, 200)[:, np.newaxis]
x_data = x_data.reshape(-1,1)
x_data = torch.FloatTensor(x_data)
y_data = md(x_data)
plt.plot(x_data, y_data.data.numpy(), 'r-')
imgurl = 'static/'+request.args['mdname']+".png"
plt.savefig(imgurl)
return render_template('loadtest.htm', url=imgurl,name="测试模型加载")
@app.route("/testtemp")
def testTemp():
return render_template('test.htm', name="测试")
if __name__ == '__main__':
app.run(host="0.0.0.0", port=80,debug = True)

文章来源：https://www.ixiera.com