Matting/main.py

from flask import Flask, request, jsonify
import os
import tools
from werkzeug.utils import secure_filename
from api import resp

app = Flask(__name__)

cur_dirs = os.path.abspath(os.path.dirname(__file__))
save_dir = os.path.join(cur_dirs, 'outputs')
upload_dir = os.path.join(cur_dirs, "uploads")

app.config['UPLOAD_FOLDER'] = upload_dir


def get_upload_file_path(name):
    return os.path.join(upload_dir, name)


@app.route("/image/seg", methods=['POST'])
def seg():
    if 'file' not in request.files:
        return '{}'
    file = request.files['file']

    if file.filename == '':
        return '{}'

    filename = secure_filename(file.filename)
    file_path = get_upload_file_path(filename)
    file.save(file_path)

    _, _, _, path = tools.seg(file_path, save_dir)
    return jsonify(resp.success_resp(path))


if __name__ == '__main__':
    tools.load_model()
    app.run(port=20201, host="0.0.0.0", debug=True)