Matting/main.py

import os

import tools
import api
from app import app

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 init():
    if os.path.exists(upload_dir) is False:
        os.makedirs(upload_dir)
    api.set_options(upload_dir, save_dir)


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