lightning-pose
Lightning Pose是一个基于PyTorch Lightning的姿态估计框架,利用NVIDIA DALI加速未标记视频的训练。框架集成了TensorBoard、FiftyOne和Streamlit等评估工具,支持已标记数据和未标记视频的模型训练与评估。项目还提供浏览器应用,涵盖数据标注、模型训练到结果可视化的完整流程。此外,团队开发的Ensemble Kalman Smoother (EKS)后处理器可与多种姿态估计包兼容。