diff --git a/ieemoo-ai-isempty.py b/ieemoo-ai-isempty.py index a7bb3d0..26eb0ba 100644 --- a/ieemoo-ai-isempty.py +++ b/ieemoo-ai-isempty.py @@ -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) diff --git a/requirements.txt b/requirements.txt index 9cc61aa..b3b7c3f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,10 +5,10 @@ matplotlib==3.4.1 numpy==1.20.2 esdk-obs-python --trusted-host pypi.org opencv_python==4.5.1.48 +opencv-contrib-python==4.5.5.64 Pillow==9.1.0 scipy==1.6.2 setuptools==49.6.0 -ml-collections==0.1.1 coremltools==5.2.0 onnx==1.7.0 pandas==1.2.4 @@ -16,5 +16,6 @@ pycocotools==2.0.2 PyYAML==6.0 requests==2.25.1 seaborn==0.11.1 +thop==0.0.31.post2005241907 tqdm==4.60.0 - +ml-collections==0.1.1