lichen/ieemoo-ai-isempty
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update ieemoo-ai-isempty.py.

Browse Source
This commit is contained in:
Brainway
2022-10-11 08:36:02 +00:00
committed by Gitee
parent e9e4dd393b
commit e2057ccf17
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
ieemoo-ai-isempty.py
View File

@ -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():