lichen/ieemoo-ai-predict
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 13:54:16 +08:00
parent cbcdb2bf8a
commit a79ed4db47
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ieemoo-ai-predict.py
View File

@ -37,7 +37,7 @@ app = Flask(__name__)
load model load model
''' '''
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-predict/', help='model.pt path(s)') parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-predict/model/now/', help='model.pt path(s)')
parser.add_argument('--Form', type=str, default='') parser.add_argument('--Form', type=str, default='')
parser.add_argument('--source', type=str, default='../module/ieemoo-ai-predict/imgs', help='source') # file/folder, 0 for webcam parser.add_argument('--source', type=str, default='../module/ieemoo-ai-predict/imgs', help='source') # file/folder, 0 for webcam
parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)') parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')

6
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,4 +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
tqdm==4.60.0 thop==0.0.31.post2005241907
tqdm==4.60.0
ml-collections==0.1.1