diff --git a/ieemoo-ai-isempty.py b/ieemoo-ai-isempty.py index 197ea00..c5815ae 100755 --- 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")