Skip to content

Latest commit

 

History

History

BoT

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
utils
utils
 
 
README.md
README.md
 
 
data_loader.py
data_loader.py
 
 
infer.sh
infer.sh
 
 
loss.py
loss.py
 
 
lr_scheduler.py
lr_scheduler.py
 
 
model.py
model.py
 
 
reid.py
reid.py
 
 
requirements.txt
requirements.txt
 
 
train.sh
train.sh
 
 

README.md

Re-ID

Introduction

This repo provides one method for Person Re-identification task (code in lazy mode).

Get started

Prerequisites

pip3 install -r requirements.txt

Data preparation

We use Market1501 to train and evaluate our Re-ID models. Please create a folder named datasets and then download it here and unzip in datasets first.

Train a model

To train a model, run sh train.sh

Our model is trained using ResNet50 as backbones

Evaluate a model

To evaluate a model, run sh infer.sh