From eca100066dc62c6146e8ddf8621449840c4545a9 Mon Sep 17 00:00:00 2001 From: lichen <770918727@qq.com> Date: Mon, 11 Apr 2022 18:05:18 +0800 Subject: [PATCH] update --- ieemoo-ai-zhanting.py | 4 +- init.sh | 2 +- requirements.txt | 97 ++++++++----------------------------------- 3 files changed, 21 insertions(+), 82 deletions(-) mode change 100755 => 100644 requirements.txt diff --git a/ieemoo-ai-zhanting.py b/ieemoo-ai-zhanting.py index 18d8749..98f6294 100755 --- a/ieemoo-ai-zhanting.py +++ b/ieemoo-ai-zhanting.py @@ -20,8 +20,8 @@ app = Flask(__name__) app.use_reloader=False 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('--source', type=str, default='data/images/1.jpg', help='source') # file/folder, 0 for webcam +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='../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.25, help='object confidence threshold') parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS') diff --git a/init.sh b/init.sh index b07ec4f..2037ecb 100644 --- a/init.sh +++ b/init.sh @@ -1,4 +1,4 @@ /opt/miniconda3/bin/conda activate ieemoo -pip3 install -r requirements.txt +/opt/miniconda3/envs/ieemoo/bin/pip install -r requirements.txt diff --git a/requirements.txt b/requirements.txt old mode 100755 new mode 100644 index 64d403c..e63de95 --- a/requirements.txt +++ b/requirements.txt @@ -1,80 +1,19 @@ -absl-py==1.0.0 -Bottleneck==1.3.2 -brotlipy==0.7.0 -cachetools==5.0.0 -certifi==2021.10.8 -cffi @ file:///tmp/build/80754af9/cffi_1625807838443/work -charset-normalizer @ file:///tmp/build/80754af9/charset-normalizer_1630003229654/work -click==8.0.3 -contextlib2==21.6.0 -cryptography @ file:///tmp/build/80754af9/cryptography_1635366571107/work -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 +coremltools==5.2.0 +Flask==2.0.0 +gevent==21.1.2 +matplotlib==3.4.1 +numpy==1.20.2 +onnx==1.7.0 +opencv_python==4.5.1.48 +pandas==1.2.4 +Pillow==9.1.0 +pycocotools==2.0.2 PyYAML==6.0 -requests @ file:///tmp/build/80754af9/requests_1629994808627/work -requests-oauthlib==1.3.1 -rsa==4.8 -scipy @ file:///tmp/build/80754af9/scipy_1630606796110/work -seaborn @ file:///tmp/build/80754af9/seaborn_1629307859561/work -sip==4.19.13 -six @ file:///tmp/build/80754af9/six_1623709665295/work -supervisor==4.2.2 -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 +requests==2.25.1 +scipy==1.6.2 +seaborn==0.11.1 +setuptools==49.6.0.post20210108 +thop==0.0.31.post2005241907 +torch==1.7.1+cu110 +torchvision==0.8.2+cu110 +tqdm==4.60.0