关键点检测

此仓库包含构建关键点检测系统的示例和最佳实践指南。它还展示了如何使用预训练模型进行人体姿态估计。

关键点定义为对象上的兴趣点。例如，人们可能对找到瓶盖在瓶子上的位置感兴趣。另一个例子是为人体姿态估计找到身体关节（手、肩膀等）。

我们使用 Mask R-CNN 的扩展，它同时检测对象及其关键点。底层技术与我们的对象检测方法非常相似，即基于 Torchvision 的 Mask R-CNN。因此，关键点定位的示例笔记本在 检测 文件夹中。

检测冰箱对象上的顶部和底部	检测牛奶瓶上的各种关键点	使用提供的预训练模型进行人体姿态估计

常见问题

请参阅对象检测文件夹中的 FAQ.md。

Notebook

Notebook 名称	描述
03_keypoint_rcnn.ipynb	此笔记本展示了如何 (i) 运行预训练模型进行人体姿态估计；以及 (ii) 训练自定义关键点检测模型。

贡献指南

请参阅根文件夹中的贡献指南。