Flask-Web-In-DL-Classifier
Flask框架在深度学习的简单应用(真的简单,纯粹扯淡)
-
pretrained-model 训练好的模型先放在checkpoint文件夹里
-
执行以下代码即可
python app.py
- 以下代码保证每次分类上传的图片时保存下来(保存在static文件夹,static很重要啊)
if request.method=="POST" and "image_data" in request.files:
file=request.files["image_data"]
filename=secure_filename(file.filename)
file.save(os.path.join("static",filename))
- 设置ip地址,端口
if __name__=="__main__":
app.run(host="10.100.110.101",port=9527,debug=True) #debug=True 保证代码能正常运行
- html 模板文件
- 图片展示
<div class="col-md-6">
<h3>Image Prediction label</h3>
<p>{{ prediction }} </p>
<body> <img src="{{url_for('static',filename=image_file)}}" width="720" height="480"></body>
</div>