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-04-11 18:05:18 +08:00
parent 9ed6e4ef13
commit eca100066d
3 changed files with 21 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ieemoo-ai-zhanting.py
View File

@ -20,8 +20,8 @@ app = Flask(__name__)
app.use_reloader=False app.use_reloader=False
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--weights', nargs='+', type=str, default='runs/zhanting/yolov5l_finetune/exp/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('--source', type=str, default='data/images/1.jpg', help='source') # file/folder, 0 for webcam 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('--img-size', type=int, default=640, help='inference size (pixels)')
parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold') parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold')
parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS') parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')

2
init.sh
View File

@ -1,4 +1,4 @@
/opt/miniconda3/bin/conda activate ieemoo /opt/miniconda3/bin/conda activate ieemoo
pip3 install -r requirements.txt /opt/miniconda3/envs/ieemoo/bin/pip install -r requirements.txt

97
requirements.txt Executable file → Normal file
View File

@ -1,80 +1,19 @@
absl-py==1.0.0 coremltools==5.2.0
Bottleneck==1.3.2 Flask==2.0.0
brotlipy==0.7.0 gevent==21.1.2
cachetools==5.0.0 matplotlib==3.4.1
certifi==2021.10.8 numpy==1.20.2
cffi @ file:///tmp/build/80754af9/cffi_1625807838443/work onnx==1.7.0
charset-normalizer @ file:///tmp/build/80754af9/charset-normalizer_1630003229654/work opencv_python==4.5.1.48
click==8.0.3 pandas==1.2.4
contextlib2==21.6.0 Pillow==9.1.0
cryptography @ file:///tmp/build/80754af9/cryptography_1635366571107/work pycocotools==2.0.2
cycler @ file:///tmp/build/80754af9/cycler_1637851556182/work
docopt==0.6.2
esdk-obs-python==3.21.8
faiss==1.7.1
Flask @ file:///tmp/build/80754af9/flask_1634118196080/work
fonttools==4.25.0
gevent @ file:///tmp/build/80754af9/gevent_1628273677693/work
google-auth==2.6.0
google-auth-oauthlib==0.4.6
greenlet @ file:///tmp/build/80754af9/greenlet_1628887725296/work
grpcio==1.44.0
gunicorn==20.1.0
h5py @ file:///tmp/build/80754af9/h5py_1637138879700/work
idna @ file:///tmp/build/80754af9/idna_1637925883363/work
importlib-metadata==4.11.3
itsdangerous @ file:///tmp/build/80754af9/itsdangerous_1621432558163/work
Jinja2 @ file:///tmp/build/80754af9/jinja2_1635780242639/work
kiwisolver @ file:///tmp/build/80754af9/kiwisolver_1612282420641/work
Markdown==3.3.6
MarkupSafe @ file:///tmp/build/80754af9/markupsafe_1621528148836/work
matplotlib @ file:///tmp/build/80754af9/matplotlib-suite_1638289681807/work
mkl-fft==1.3.1
mkl-random @ file:///tmp/build/80754af9/mkl_random_1626186064646/work
mkl-service==2.4.0
ml-collections==0.1.0
munkres==1.1.4
numexpr @ file:///tmp/build/80754af9/numexpr_1618856167419/work
numpy @ file:///tmp/build/80754af9/numpy_and_numpy_base_1634095647912/work
oauthlib==3.2.0
olefile @ file:///Users/ktietz/demo/mc3/conda-bld/olefile_1629805411829/work
opencv-python==4.5.4.60
packaging @ file:///tmp/build/80754af9/packaging_1637314298585/work
pandas==1.3.4
Pillow==8.4.0
pipreqs==0.4.11
protobuf==3.19.4
pyasn1==0.4.8
pyasn1-modules==0.2.8
pycparser @ file:///tmp/build/80754af9/pycparser_1636541352034/work
pycryptodome==3.10.1
pyOpenSSL @ file:///tmp/build/80754af9/pyopenssl_1635333100036/work
pyparsing @ file:///tmp/build/80754af9/pyparsing_1635766073266/work
PySocks @ file:///tmp/build/80754af9/pysocks_1605305779399/work
python-dateutil @ file:///tmp/build/80754af9/python-dateutil_1626374649649/work
pytz==2021.3
PyYAML==6.0 PyYAML==6.0
requests @ file:///tmp/build/80754af9/requests_1629994808627/work requests==2.25.1
requests-oauthlib==1.3.1 scipy==1.6.2
rsa==4.8 seaborn==0.11.1
scipy @ file:///tmp/build/80754af9/scipy_1630606796110/work setuptools==49.6.0.post20210108
seaborn @ file:///tmp/build/80754af9/seaborn_1629307859561/work thop==0.0.31.post2005241907
sip==4.19.13 torch==1.7.1+cu110
six @ file:///tmp/build/80754af9/six_1623709665295/work torchvision==0.8.2+cu110
supervisor==4.2.2 tqdm==4.60.0
tensorboard==2.8.0
tensorboard-data-server==0.6.1
tensorboard-plugin-wit==1.8.1
torch==1.8.0
torchaudio==0.8.0a0+a751e1d
torchvision==0.9.0
tornado @ file:///tmp/build/80754af9/tornado_1606942300299/work
tqdm @ file:///tmp/build/80754af9/tqdm_1635330843403/work
typing-extensions @ file:///tmp/build/80754af9/typing_extensions_1631814937681/work
urllib3==1.26.7
Werkzeug @ file:///tmp/build/80754af9/werkzeug_1635505089296/work
yacs @ file:///tmp/build/80754af9/yacs_1634047592950/work
yarg==0.1.9
zipp==3.7.0
zope.event==4.5.0
zope.interface @ file:///tmp/build/80754af9/zope.interface_1625035545636/work