花一個小時，用Flask寫一個簡單的行人檢測介面
- 前提條件
- 實驗環境
- 專案結構
- 主要代碼
- 運行結果
- 參考鏈接
前提條件
- 了解Python語言，并會安裝第三方庫
- 了解Python Web Flask框架
- 了解PyTorch深度學習框架
實驗環境
- Python 3.6.2
- PyTorch 1.7.1
- Flask 1.1.1
- Numpy 1.18.5
- Opencv 3.4.2
- PIL
pip3 install pillow
專案結構
相關說明：
- static：用于存盤靜態檔案，比如css、js和圖片等
- templates：存放模板檔案
- upload：用于保存上傳檔案
- flask_app.py： 應用程式主檔案
- predict.py：預測檔案
主要代碼
完整代碼，暫時沒空整理，如整理完，后續會發布，敬請期待！
#!/usr/bin/python
# -*- coding: UTF-8 -*-
import imp
from flask import request, jsonify, send_from_directory, abort
from werkzeug.utils import secure_filename
from flask import Flask, render_template, jsonify, request
from predict import pre
import time
import os
import base64
app = Flask(__name__)
UPLOAD_FOLDER = 'upload'
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
basedir = os.path.abspath(os.path.dirname(__file__))
ALLOWED_EXTENSIONS = set(['txt', 'png', 'jpg', 'xls', 'JPG', 'PNG', 'xlsx', 'gif', 'GIF'])
# 用于判斷檔案后綴
def allowed_file(filename):
return '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS
# 上傳
@app.route('/upload')
def upload_test():
return render_template('upload.html')
@app.route("/api/download/<filename>", methods=['GET'])
def download(filename):
if request.method == "GET":
if os.path.isfile(os.path.join('upload', filename)):
return send_from_directory('upload', filename, as_attachment=True)
abort(404)
# 上傳檔案
@app.route('/api/upload', methods=['POST'], strict_slashes=False)
def api_upload():
file_dir = os.path.join(basedir, app.config['UPLOAD_FOLDER'])
if not os.path.exists(file_dir):
os.makedirs(file_dir)
f = request.files['myfile'] # 從表單的file欄位獲取檔案，myfile為該表單的name值
if f and allowed_file(f.filename): # 判斷是否是允許上傳的檔案型別
fname = secure_filename(f.filename)
print(fname)
ext = fname.rsplit('.', 1)[1] # 獲取檔案后綴
unix_time = int(time.time())
new_filename = str(unix_time) + '.' + ext # 修改了上傳的檔案名
f.save(os.path.join(file_dir, new_filename)) # 保存檔案到upload目錄
img_path = os.path.join("upload", new_filename)
print(img_path)
pre_result = pre(img_path)
print(pre_result)
token = base64.b64encode(new_filename.encode('utf-8'))
print(token)
return jsonify({"code": 0, "errmsg": "OK", "token": token, "fileName": "/api/download/" + new_filename,"detect_result:":pre_result})
else:
return jsonify({"code": 1001, "errmsg": "ERROR"})
if __name__ == '__main__':
app.run(host="0.0.0.0",port="5000",threaded=True,debug=False)
<!DOCTYPE html>
<html>
<head>
<meta charset="UTF-8">
<link href="{{url_for('static', filename='obj_classification.css')}}" rel="stylesheet" type="text/css" />
<title>圖片識別--Person</title>
</head>
<body>
<h1>圖片識別--Person</h1>
<div class="container">
<div class="choose">
<form action="http://IP地址:5000/api/upload" enctype='multipart/form-data' method='POST'>
<input type="file" name="myfile" class="input-new" style="margin-top:20px;" />
<input type="submit" value="識別圖片" class="button-new" style="margin-top:15px;" />
</form>
</div>
<div class="display">
<img src="{{ url_for('static', filename='images/test.jpg',_t=val1) }}" width="400" height="500" alt="圖片" />
</div>
</div>
</body>
</html>
運行結果
{
"code": 0,
"detect_result:": [
{
"bbox": [
51.0,
265.0,
543.0,
437.0
],
"class": "b'person 0.78'"
},
{
"bbox": [
43.0,
433.0,
543.0,
609.0
],
"class": "b'person 0.77'"
},
{
"bbox": [
44.0,
133.0,
543.0,
309.0
],
"class": "b'person 0.76'"
},
{
"bbox": [
46.0,
526.0,
543.0,
665.0
],
"class": "b'person 0.74'"
},
{
"bbox": [
107.0,
51.0,
525.0,
181.0
],
"class": "b'person 0.62'"
}
],
"errmsg": "OK",
"fileName": "/api/download/1645974252.jpg",
"token": "MTY0NTk3NDI1Mi5qcGc="
}
參考鏈接
https://flask.palletsprojects.com/en/2.0.x/
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/434536.html
標籤：AI