MindPose姿态估计工具箱

MindPose 是一个开源的基于 MindSpore 的姿态估计工具箱。它收集了一系列经典和 SoTA 视觉模型，例如 HRNet，以及它们的预训练权重和训练策略。

主要特点

• 便于使用。MindPose 将视觉框架分解为各种可配置的组件。使用 MindPose 可以轻松自定义你的数据管道、模型和学习管道：

>>>  import  mindpose  # 创建模型 >>>  network  =  mindpose . create_network ( backbone_name = 'resnet50' ,  head_name = "simple_baseline_head" )

• 最先进的。MindPose 提供了各种基于 CNN 和 Transformer 的视觉模型。他们提供预训练的权重和性能报告，以帮助用户选择和重用正确的模型。

• 灵活性和效率。MindPose 建立在 MindSpore 之上，MindSpore 是一个高效的 DL 框架，可以在不同的硬件平台（GPU/CPU/Ascend）上运行。它支持高效的图形模式和灵活的拼写模式。