update ieemoo-ai-isempty.py.

2022-09-28 04:05:00 +00:00
parent 7a1d066fdc
commit 23b382e459
1 changed files with 1 additions and 1 deletions
--- a/ieemoo-ai-isempty.py
+++ b/ieemoo-ai-isempty.py
@ -40,7 +40,7 @@ app = Flask(__name__)
 app.use_reloader=False


-def parse_args(model_file="./output/ieemooempty_checkpoint_good.pth"):
+def parse_args(model_file="../module/ieemoo-ai-isempty/model/now/emptyjudge5_checkpoint.bin"):
    parser = argparse.ArgumentParser()
    parser.add_argument("--img_size", default=600, type=int, help="Resolution size")
    parser.add_argument('--split', type=str, default='overlap', help="Split method")