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

update

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
ieemoo-ai-isempty.py
View File

@ -18,7 +18,7 @@ app = Flask(__name__)
app.use_reloader=False 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"): #def parse_args(model_file="output/emptyjudge5_checkpoint.bin"):
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--img_size", default=448, type=int, help="Resolution size") parser.add_argument("--img_size", default=448, type=int, help="Resolution size")
@ -85,7 +85,7 @@ class Predictor(object):
return result 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" #model_file ="output/emptyjudge5_checkpoint.bin"
args = parse_args(model_file) args = parse_args(model_file)
predictor = Predictor(args) predictor = Predictor(args)

5
requirements.txt
View File

@ -5,10 +5,10 @@ matplotlib==3.4.1
numpy==1.20.2 numpy==1.20.2
esdk-obs-python --trusted-host pypi.org esdk-obs-python --trusted-host pypi.org
opencv_python==4.5.1.48 opencv_python==4.5.1.48
opencv-contrib-python==4.5.5.64
Pillow==9.1.0 Pillow==9.1.0
scipy==1.6.2 scipy==1.6.2
setuptools==49.6.0 setuptools==49.6.0
ml-collections==0.1.1
coremltools==5.2.0 coremltools==5.2.0
onnx==1.7.0 onnx==1.7.0
pandas==1.2.4 pandas==1.2.4
@ -16,5 +16,6 @@ pycocotools==2.0.2
PyYAML==6.0 PyYAML==6.0
requests==2.25.1 requests==2.25.1
seaborn==0.11.1 seaborn==0.11.1
thop==0.0.31.post2005241907
tqdm==4.60.0 tqdm==4.60.0
ml-collections==0.1.1