MODEL_ZOO.md

CenterGroup Checkpoints

The following checkpoints correspond to our final models:

Train Data	Method	Detector	Input size	Val/Test AP	Config	Checkpoint
COCO Train 2017	CenterGroup	HigherHRNet-w32	512	69.0	cfg	ckpt
COCO Train 2017	CenterGroup	HigherHRNet-w48	640	71.0	cfg	ckpt
CrowdPose train+val	CenterGroup	HigherHRNet-w48	640	67.6	cfg	ckpt

Note that we report single-scale AP on the respective val/test dataset (val for COCO, test for CrowdPose).

HigherHRNet with Centers

As explained in the paper, before training CenterGroup, we first pretrain our Keypoint Detector, HigherHRNet, by adding an additional keypoint prediction corresponding to person centers. When training CenterGroup, these checkpoints are needed to initialize our detector.

Train Data	Method	Detector	Input size	Test AP	Config	Checkpoint
COCO Train 2017	CenterGroup	HigherHRNet-w32	512	67.2	cfg	ckpt
COCO Train 2017	CenterGroup	HigherHRNet-w48	640	69.7	cfg	ckpt
CrowdPose train+val	CenterGroup	HigherHRNet-w48	640	??	cfg	ckpt

Note that we report single-scale AP on the respective val/test dataset (val for COCO, test for CrowdPose).

Our config files expect to find these checkpoints under $CENTERGROUP_ROOT/models. Please update the config files if you download them somewhere else.