diff --git a/ieemoo-ai-isempty.py b/ieemoo-ai-isempty.py index a0a9da2..1313fdf 100755 --- a/ieemoo-ai-isempty.py +++ b/ieemoo-ai-isempty.py @@ -38,21 +38,21 @@ sys.path.insert(0, ".") app = Flask(__name__) -app.use_reloader=False +#app.use_reloader=False print("Autor:ieemoo_lx") -def parse_args(model_file="./output/ieemooempty_vit_checkpoint.pth"): +def parse_args(): 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") parser.add_argument('--slide_step', type=int, default=2, help="Slide step for overlap split") parser.add_argument('--smoothing_value', type=float, default=0.0, help="Label smoothing value") - parser.add_argument("--pretrained_model", type=str, default=model_file, help="load pretrained model") + parser.add_argument("--pretrained_model", type=str, default="../module/ieemoo-ai-isempty/model/now/emptyjudge5_checkpoint.bin", help="load pretrained model") + #parser.add_argument("--pretrained_model", type=str, default="./output/ieemooempty_vit_checkpoint.pth", help="load pretrained model") opt, unknown = parser.parse_known_args() return opt - class Predictor(object): def __init__(self, args): self.args = args @@ -107,10 +107,9 @@ class Predictor(object): return result -model_file ="./output/ieemooempty_checkpoint_good.pth" -args = parse_args(model_file) -predictor = Predictor(args) +args = parse_args() +predictor = Predictor(args) @app.route("/isempty", methods=['POST']) def get_isempty():