Code for CVPR 2021 paper: Anchor-Free Person Search

Introduction

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

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

2. Test CUHK-SYSU

Change the paths in L59 and L72 in test_results.py

sh run_test.sh

3. Test PRW

Change the paths in L127 and L128 in test_results_prw.py

sh run_test_prw.sh

Performance

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}
}