This commit is contained in:
lichen
2022-05-05 11:54:40 +08:00
parent c8d85a9f03
commit 0b34aa05e0
2 changed files with 5 additions and 4 deletions

View File

@ -18,7 +18,7 @@ app = Flask(__name__)
app.use_reloader=False
def parse_args(model_file="../module/ieemoo-ai-isempty/emptyjudge5_checkpoint.bin"):
def parse_args(model_file="../module/ieemoo-ai-isempty/model/now/emptyjudge5_checkpoint.bin"):
#def parse_args(model_file="output/emptyjudge5_checkpoint.bin"):
parser = argparse.ArgumentParser()
parser.add_argument("--img_size", default=448, type=int, help="Resolution size")
@ -85,7 +85,7 @@ class Predictor(object):
return result
model_file ="../module/ieemoo-ai-isempty/emptyjudge5_checkpoint.bin"
model_file ="../module/ieemoo-ai-isempty/model/now/emptyjudge5_checkpoint.bin"
#model_file ="output/emptyjudge5_checkpoint.bin"
args = parse_args(model_file)
predictor = Predictor(args)