|
@@ -1,5 +1,6 @@
|
|
|
+import argparse
|
|
|
import os
|
|
|
-
|
|
|
+import sys
|
|
|
import tools
|
|
|
import api
|
|
|
from app import app
|
|
@@ -17,7 +18,21 @@ def init():
|
|
|
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__':
|
|
|
+ args = parse_args()
|
|
|
init()
|
|
|
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)
|