lichen/ieemoo-ai-zhanting
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:44:14 +08:00
parent 41f69858ce
commit 9270888194
2 changed files with 4 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ieemoo-ai-zhanting.py
View File

@ -21,7 +21,7 @@ app.use_reloader=False
parser = argparse.ArgumentParser()
#parser.add_argument('--weights', nargs='+', type=str, default='runs/zhanting/yolov5m_finetune/exp4/weights/best.pt', help='model.pt path(s)')
parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-zhanting/best.pt', help='model.pt path(s)')
parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-zhanting/model/now/best.pt', help='model.pt path(s)')
parser.add_argument('--source', type=str, default='../module/ieemoo-ai-zhanting/imgs/1.jpg', help='source') # file/folder, 0 for webcam
parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
parser.add_argument('--conf-thres', type=float, default=0.50, help='object confidence threshold')

13
requirements.txt
View File

@ -5,17 +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
--find-links https://download.pytorch.org/whl/torch_stable.html
torch==1.8.0+cu111
--find-links https://download.pytorch.org/whl/torch_stable.html
torchvision==0.9.0+cu111
--find-links https://download.pytorch.org/whl/torch_stable.html
torchaudio==0.8.0
coremltools==5.2.0
onnx==1.7.0
pandas==1.2.4
@ -23,6 +16,6 @@ pycocotools==2.0.2
PyYAML==6.0
requests==2.25.1
seaborn==0.11.1
thop==0.0.31
thop==0.0.31.post2005241907
tqdm==4.60.0
ml-collections==0.1.1