diff --git a/.gitignore b/.gitignore index a80119f..fce1a35 100644 --- a/.gitignore +++ b/.gitignore @@ -1,4 +1,5 @@ .env/ +.venv/ **__pycache__** *.onnx *.pt @@ -13,4 +14,7 @@ asone.egg-info/ test_custom_tracker.py custom_weights.py data/results/ -temp/ \ No newline at end of file +temp/ +test* +yolo* +*.mp4 \ No newline at end of file diff --git a/README.md b/README.md index c26d537..f37cc59 100644 --- a/README.md +++ b/README.md @@ -56,11 +56,8 @@ Change Directory to AS-One python3 -m venv .env source .env/bin/activate -pip install numpy Cython -pip install cython-bbox asone onnxruntime-gpu==1.12.1 -pip install super-gradients==3.1.1 -# for CPU -pip install torch torchvision +pip install -r requirements.txt + ``` diff --git a/main.py b/main.py index 1788cc6..be930df 100644 --- a/main.py +++ b/main.py @@ -10,7 +10,7 @@ def main(args): detect = ASOne( tracker=asone.BYTETRACK, - detector=asone.YOLOV7_PYTORCH, + detector=asone.YOLONAS_S_PYTORCH, weights=args.weights, use_cuda=args.use_cuda ) diff --git a/requirements.txt b/requirements.txt index f72aaba..e0ba125 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,28 +1,175 @@ -lap -loguru -norfair -numpy==1.23.3 -#onnxruntime-gpu==1.12.1 -opencv-python -scipy -pyyaml -easydict -gdown -pandas -tabulate -typing-extensions==3.10.0.2 -wheel -torch==1.12.1 -torchvision -numpy==1.23.3 -Cython -ultralytics -asone-ocr -motpy - +numpy==1.23.0 +Cython==0.29.34 +cython-bbox==0.1.3 +absl-py==1.4.0 +alabaster==0.7.13 +antlr4-python3-runtime==4.9.3 +asone-ocr==1.6.2 +asttokens==2.2.1 +attrs==23.1.0 +Babel==2.12.1 +backcall==0.2.0 +beautifulsoup4==4.12.2 +boto3==1.26.133 +botocore==1.29.133 +build==0.10.0 +cachetools==5.3.0 +certifi==2023.5.7 +charset-normalizer==3.1.0 +click==8.1.3 +cmake==3.26.3 +coloredlogs==15.0.1 +comm==0.1.3 +commonmark==0.9.1 +contourpy==1.0.7 +coremltools==6.3.0 +coverage==5.3.1 +cycler==0.11.0 +debugpy==1.6.7 +decorator==5.1.1 +Deprecated==1.2.13 +docutils==0.18.1 +easydict==1.10 +einops==0.3.2 +executing==1.2.0 +filelock==3.12.0 +filterpy==1.4.5 +flatbuffers==23.5.9 +fonttools==4.39.4 +future==0.18.3 +gdown==4.7.1 +google-auth==2.18.0 +google-auth-oauthlib==1.0.0 +grpcio==1.54.0 +humanfriendly==10.0 +hydra-core==1.3.2 +idna==3.4 +imageio==2.28.1 +imagesize==1.4.1 +importlib-metadata==6.6.0 +importlib-resources==5.12.0 +ipykernel==6.23.0 +ipython==8.12.2 +jedi==0.18.2 +Jinja2==3.1.2 +jmespath==1.0.1 +json-tricks==3.16.1 +jsonschema==4.17.3 +jupyter_client==8.2.0 +jupyter_core==5.3.0 +kiwisolver==1.4.4 +lap==0.4.0 +lazy_loader==0.2 +lit==16.0.3 +loguru==0.7.0 +Markdown==3.4.3 +MarkupSafe==2.1.2 +matplotlib==3.7.1 +matplotlib-inline==0.1.6 +motpy==0.0.10 +mpmath==1.3.0 +nest-asyncio==1.5.6 +networkx==3.1 +ninja==1.11.1 +norfair==2.2.0 +nvidia-cublas-cu11==11.10.3.66 +nvidia-cuda-cupti-cu11==11.7.101 +nvidia-cuda-nvrtc-cu11==11.7.99 +nvidia-cuda-runtime-cu11==11.7.99 +nvidia-cudnn-cu11==8.5.0.96 +nvidia-cufft-cu11==10.9.0.58 +nvidia-curand-cu11==10.2.10.91 +nvidia-cusolver-cu11==11.4.0.1 +nvidia-cusparse-cu11==11.7.4.91 +nvidia-nccl-cu11==2.14.3 +nvidia-nvtx-cu11==11.7.91 +oauthlib==3.2.2 +omegaconf==2.3.0 +onnx==1.13.0 +onnx-simplifier==0.4.28 +onnxruntime==1.13.1 +onnxruntime-gpu==1.12.1 +opencv-python==4.7.0.72 +packaging==23.1 +pandas==2.0.1 +parso==0.8.3 +pexpect==4.8.0 +pickleshare==0.7.5 +Pillow==9.5.0 +pip-tools==6.13.0 +pkg_resources==0.0.0 +pkgutil_resolve_name==1.3.10 +platformdirs==3.5.1 +prompt-toolkit==3.0.38 +protobuf==3.20.3 +psutil==5.9.5 +ptyprocess==0.7.0 +pure-eval==0.2.2 +pyasn1==0.5.0 +pyasn1-modules==0.3.0 +pyclipper==1.3.0.post4 +pycocotools==2.0.4 +pyDeprecate==0.3.2 +Pygments==2.15.1 +pyparsing==2.4.5 +pyproject_hooks==1.0.0 +pyrsistent==0.19.3 +PySocks==1.7.1 +python-bidi==0.4.2 +python-dateutil==2.8.2 +pytz==2023.3 +PyWavelets==1.4.1 +PyYAML==6.0 +pyzmq==25.0.2 +rapidfuzz==3.0.0 +requests==2.30.0 +requests-oauthlib==1.3.1 +rich==12.6.0 +rsa==4.9 +s3transfer==0.6.1 +scikit-image==0.20.0 +scipy==1.10.1 +seaborn==0.12.2 +sentry-sdk==1.22.2 +shapely==2.0.1 +six==1.16.0 +snowballstemmer==2.2.0 +soupsieve==2.4.1 +Sphinx==4.0.3 +sphinx-rtd-theme==1.2.0 +sphinxcontrib-applehelp==1.0.4 +sphinxcontrib-devhelp==1.0.2 +sphinxcontrib-htmlhelp==2.0.1 +sphinxcontrib-jquery==4.1 +sphinxcontrib-jsmath==1.0.1 +sphinxcontrib-qthelp==1.0.3 +sphinxcontrib-serializinghtml==1.1.5 +stack-data==0.6.2 +stringcase==1.2.0 +super-gradients==3.1.1 +sympy==1.12 +tabulate==0.9.0 +tensorboard==2.13.0 +tensorboard-data-server==0.7.0 +termcolor==1.1.0 +thop==0.1.1.post2209072238 +tifffile==2023.4.12 +tomli==2.0.1 +torch==2.0.1 +torchaudio==2.0.2 +torchmetrics==0.8.0 torchreid==0.2.5 -tensorboard -protobuf==3.20 -# requirements for coreml -onnxruntime -coremltools +torchvision==0.15.2 +tornado==6.3.1 +tqdm==4.65.0 +traitlets==5.9.0 +treelib==1.6.1 +triton==2.0.0 +typing-extensions==3.10.0.2 +tzdata==2023.3 +ultralytics==8.0.98 +urllib3==1.26.15 +wcwidth==0.2.6 +Werkzeug==2.3.4 +wrapt==1.15.0 +zipp==3.15.0