/AlignPS

Code for CVPR 2021 paper: Anchor-Free Person Search

Primary LanguagePythonApache License 2.0Apache-2.0

Introduction

This is the implementationn for Anchor-Free Person Search in CVPR2021

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.

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, Jingpeng Li, Jie Qin, Song Bai, Shengcai Liao, Li Liu, Fan Zhu, Ling Shao},
  booktitle={CVPR},
  year={2021}
}