pytorch-cpn

收录理由

做姿态估计的朋友想找一个能快速读完的代码库，这个PyTorch版级联金字塔网络值得一看。原模型拿过2017年MSCOCO关键点挑战赛冠军，而这份复现把设计思路保持得很清晰：GlobalNet负责容易检测的关节点，RefineNet再回头处理难定位的部分，两阶段的想法没有被框架细节掩盖。仓库里带了完整的COCO训练和验证流程，还有ResNet-50和ResNet-101在不同输入尺寸下的AP结果，想复现的话可以先对照着检查。对于刚接触多人关键点估计的人，一个体量小、读得懂的代码加上公开的评测数字，比一头扎进庞大的工业级框架要友好得多。