• Install Pytorch from http://pytorch.org/
• Install Torchvision from the source

git clone https://github.com/pytorch/vision
cd vision
python setup.py install

• [Optinal] You may skip it. Install apex from the source

git clone https://github.com/NVIDIA/apex.git
cd apex
python setup.py install --cuda_ext --cpp_ext

Because pytorch and torchvision are ongoing projects.

Here we noted that our code is tested based on Pytorch 1.10.0 and CUDA 11.3 on Windows 11.

Download Market1501 Dataset [Google] [Baidu]

Preparation: Put the images with the same id in one folder. You may use

python prepare.py

Remember to change the dataset path to your own path.

Just following HRNet-reID.ipynb~