MODEL ZOO

August 15, 2020 ยท View on GitHub

MSCOCO dataset

ModelBackboneDetectorInput SizeAPSpeedDownloadConfigTraining Log
Simple BaselineResNet50YOLOv3256x19270.62.94 iter/smodelcfglog
Fast PoseResNet50YOLOv3256x19272.03.54 iter/smodelcfglog
Fast Pose (DUC)ResNet50 - unshuffleYOLOv3256x19272.42.91 iter/smodelcfglog
HRNetHRNet-W32YOLOv3256x19272.52.13 iter/smodelcfglog
Fast Pose (DCN)ResNet50 - dcnYOLOv3256x19272.82.94 iter/smodelcfglog
Fast Pose (DUC)ResNet152YOLOv3256x19273.31.62 iter/smodelcfglog

Notes

  • All models are trained on keypoint train 2017 images which contains at least one human with keypoint annotations (64115 images).
  • The evaluation is done on COCO keypoint val 2017 (5000 images).
  • Flip test is used by default.
  • One TITAN XP is used for speed test, with batch_size=64 in each iteration.
  • Offline human detection results are used in speed test.
  • FastPose is our own network design. Paper coming soon!

Halpe dataset (26 keypoints)

ModelBackboneDetectorInput SizeAPSpeedDownloadConfigTraining Log
Fast PoseResNet50YOLOv3256x19269.03.54 iter/sGoogle Baiducfglog

Notes

  • More models coming soon!

Halpe dataset (136 keypoints)

ModelBackboneDetectorInput SizeAPSpeedDownloadConfigTraining Log
Fast PoseResNet50YOLOv3256x19269.03.54 iter/sGoogle Baiducfglog

Notes

  • Coming soon!