peterzs / light-reid Goto Github PK

This project forked from wangguanan/light-reid

[ECCV2020] a toolbox of light-reid learning for faster inference, speed both feature extraction and retrieval stages up to >30x

Python 100.00%

light-reid's Introduction

light-reid

a toolbox of light reid for fast feature extraction and search

light-model: model distillation (3x faster feature extraction)
light-feature: binary code learning (6x faster retrieval)
light-search: coarse2fine search (2x faster retrieval)

it features

easy switch between light and non-light reid
simple modules for reid implementation
implementations of state-of-the-art deep reid models

What's New

[2020.08]: We release a toolbox of light-reid learning for faster inference, getting >30x faster speed.
[2020.03]: We implement BagTricks and support IBN-Net, MSMT17, combineall, multi-dataset train. Please see branch version_py3.7_bot.
[2019.03]: We give a clearn implemention of BagTricks with python2.7. Please see branch version_py2.7.

Find our Works

[2020.07]: [ECCV'20] Our work about Fast ReID has been accepted by ECCV'20. ([Paper] Code)
[2020.03]: [CVPR'20] Our work about Occluded ReID has been accepted by CVPR'20. (Paper, Code).
[2020.01]: [AAAI'20] Our work about RGB-Infrared(IR) ReID has been accepted by AAAI'20. (Paper, Code).
[2019.10]: [ICCV'19] Our work about RGB-Infrared(IR) ReID has been accepted by ICCV'19. (Paper, Code).
[2019.05]: We implement PCB and achieve better performance than the offical one. (Code)

Installation

# clone this repo
git clone https://github.com/wangguanan/light-reid.git

# create environment
cd light-reid
conda create -n lightreid python=3.7
conda activate lightreid

# install dependencies
pip install -r requirements.txt

# install torch and torchvision (select the proper cuda version to suit your machine)
conda install pytorch==1.4.0 torchvision -c pytorch