Customized Torchreid for pyppbox

Torchreid is a library for deep-learning person re-identification using PyTorch, and pyppbox-torchreid is a customized Torchreid for pyppbox and:

• Ensures that Cython natively works on all OS platforms (Windows/Linux/macOS),
• 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 the pre-built wheel in releases or install from PyPI:

pip install pyppbox-torchreid

Other options

Install from GitHub repo (Require C++ compiler):

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

Build and install (Require C++ compiler):

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

🔬 Test

Make sure you install pyppbox-torchred, OpenCV, PyTorch and other requirements; for example, with GPU support on Windows:

pip install opencv-contrib-python
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
pip install pyppbox-torchreid

Test Cython rank_cylib

git clone https://github.com/rathaumons/torchreid-for-pyppbox.git
cd pyppbox_torchreid/metrics/rank_cylib
python test_cython.py