瀏覽代碼

启动参数

tuon 1 年之前
父節點
當前提交
004b1669d8
共有 2 個文件被更改,包括 19 次插入3 次删除
  1. 2 1
      app.py
  2. 17 2
      main.py

+ 2 - 1
app.py

@@ -1,4 +1,5 @@
 from flask import Flask
 from flask import Flask
 
 
 
 
-app = Flask(__name__)
+print("name: {}".format(__name__))
+app = Flask("Matting")

+ 17 - 2
main.py

@@ -1,5 +1,6 @@
+import argparse
 import os
 import os
-
+import sys
 import tools
 import tools
 import api
 import api
 from app import app
 from app import app
@@ -17,7 +18,21 @@ def init():
     api.set_options(upload_dir, save_dir)
     api.set_options(upload_dir, save_dir)
 
 
 
 
+def parse_args():
+    parser = argparse.ArgumentParser(
+        description='PP-HumanSeg inference for video')
+    parser.add_argument(
+        "--debug",
+        dest="debug",
+        help="debug mode",
+        default=True,
+        type=bool,
+        required=False)
+    return parser.parse_args()
+
+
 if __name__ == '__main__':
 if __name__ == '__main__':
+    args = parse_args()
     init()
     init()
     tools.load_model()
     tools.load_model()
-    app.run(port=20201, host="0.0.0.0", debug=True)
+    app.run(port=20201, host="0.0.0.0", debug=args.debug)