update
This commit is contained in:
@ -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)
|
||||
|
Reference in New Issue
Block a user