一、引入干系模块及模块详解
引入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,requestimport torchimport jsonimport numpy as npimport matplotlib.pyplot as pltapp = 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
理解更多
