pyppbox-torchreid 1.4.0.1

Customized Torchreid for pyppbox: Deep learning person re-identification.

Customized Torchreid for pyppbox

Torchreid is a library for deep-learning person re-identification using PyTorch.

pyppbox-torchreid is a customized of Torchreid for pyppbox and:

• Ensures that Cython natively works on all OS platform (Windows/Linux),
• Enables freedom of passing local model/weight files from anywhere,
• Disables some models which are not used in pyppbox.

All source credit and more info -> Original KaiyangZhou's repo.

Install

Use Wheels fron releases or directly install from PyPI:

pip install pyppbox-torchreid

Or install from GitHub directly:

pip install git+https://github.com/rathaumons/torchreid-for-pyppbox.git

To be able to run, you must install OpenCV and PyTorch with GPU:

pip install opencv-contrib-python
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

Note: PyTorch doesn't support GUP on macOS.

Build from source

(Optional, auto install) Building Wheels or source distribution only requires these modules:

pip install "setuptools>=67.2.0"
pip install "Cython>=0.29.32"
pip install "numpy>=1.21.6"

Recommend using build for building both .whl and .tar.gz:

git clone https://github.com/rathaumons/torchreid-for-pyppbox/
cd torchreid-for-pyppbox
pip install wheel build
python -m build --sdist
python -m build --wheel
cd dist

After you install pyppbox-torchred, OpenCV and PyTorch, you can check if Cython rank_cy works:

cd pyppbox_torchreid/metrics/rank_cylib
python test_cython.py