[2023-12-27] Flask #1:Flask简单使用

Flask快速上手

环境下载

pip install flask

Flask服务(Flask Server)

from flask import Flask, request, send_file, make_response
from model import *
import os
import torch
import cv2

# Use CUDA
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
os.environ['CUDA_LAUNCH_BLOCKING'] = '1'
use_cuda = torch.cuda.is_available()

app = Flask(__name__)

# 载入模型
model=...
root_dir = ...
if not os.path.exists(root_dir):
    os.makedirs(root_dir)

# 定义服务接口
@app.route('/predict_from_file', methods=['POST'])
def predict():
    file = request.files.get('file')  # 获取上传的文件
    if file:
        file.save(root_dir + '/' + file.filename)  # 将上传文件保存到本地
        input_file = load(root_dir + '/' + file.filename)  # 读取本地文件
        results = model.inference(input_file)  # 对本地文件进行推理计算
        # 返回预测结果
        return make_response(results)
    else:
        return 'No file uploaded'


# 定义服务接口
@app.route('/get_file', methods=['GET'])
def getresult():
    filename = request.args.get('file')  # 获取请求参数中的文件名
    if not filename:
        return "Missing parameter: file"  # 没有提供文件名
    filepath = root_dir + '/' + filename  # 生成完整的文件路径
    try:
        return send_file(filepath, as_attachment=True, attachment_filename=filename)
    except FileNotFoundError:
        return "The file does not exist"  # 文件不存在

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=8000)

Flask客户端(Flask Client)