correct dependencies

.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/
+temp/
+test*
+yolo*
+*.mp4

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
+
 ```
 </details>
 

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
         )

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