/AlignPS

Code for CVPR 2021 paper: Anchor-Free Person Search

Primary LanguagePythonApache License 2.0Apache-2.0

Introduction

This is the implementation for Anchor-Free Person Search in CVPR2021.

A brief introduction in Chinese can be found at https://zhuanlan.zhihu.com/p/359617800

demo image

License

This project is released under the Apache 2.0 license.

Installation

This project is developed upon MMdetection, please refer to install.md to install MMdetection.

We utilized mmcv=1.1.5, pytorch=1.7.0

Dataset

Download CUHK-SYSU and PRW.

We provide coco-style annotation in demo/anno.

For CUHK-SYSU, change the path of your dataset and the annotaion file in the config file L3, L38, L43, L48

For PRW, change the paths in these config files: config1 config2

Experiments

  1. Train
sh run_train.sh
  1. Test CUHK-SYSU

Change the paths in L59 and L72 in test_results.py

sh run_test.sh
  1. Test PRW

Change the paths in L127 and L128 in test_results_prw.py

sh run_test_prw.sh

Performance

Dataset Model mAP Rank1 Config Link
CUHK-SYSU AlignPS 93.1% 93.4% cfg model
CUHK-SYSU AlignPS+ 94.0% 94.5% cfg model
PRW AlignPS 45.9% 81.9% cfg model
PRW AlignPS+ 46.1% 82.1% cfg model

Citation

If you use this toolbox or benchmark in your research, please cite this project.

@inproceedings{yan2021alignps,
  title={Anchor-Free Person Search},
  author={Yichao Yan, Jinpeng Li, Jie Qin, Song Bai, Shengcai Liao, Li Liu, Fan Zhu, Ling Shao},
  booktitle={CVPR},
  year={2021}
}