MindPose姿態(tài)估計(jì)工具箱

MindPose 是一個(gè)開源的基于 MindSpore 的姿態(tài)估計(jì)工具箱。它收集了一系列經(jīng)典和 SoTA 視覺模型，例如 HRNet，以及它們的預(yù)訓(xùn)練權(quán)重和訓(xùn)練策略。

主要特點(diǎn)

• 便于使用。MindPose 將視覺框架分解為各種可配置的組件。使用 MindPose 可以輕松自定義你的數(shù)據(jù)管道、模型和學(xué)習(xí)管道：

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

• 最先進(jìn)的。MindPose 提供了各種基于 CNN 和 Transformer 的視覺模型。他們提供預(yù)訓(xùn)練的權(quán)重和性能報(bào)告，以幫助用戶選擇和重用正確的模型。

• 靈活性和效率。MindPose 建立在 MindSpore 之上，MindSpore 是一個(gè)高效的 DL 框架，可以在不同的硬件平臺(tái)（GPU/CPU/Ascend）上運(yùn)行。它支持高效的圖形模式和靈活的拼寫模式。