
PoseNFS: 基于 PyTorch 的 Pose Neural Fabrics Search (PNFS) 用于人体姿态估计...
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
PoseNFS是一款基于PyTorch框架开发的人体姿态估计算法工具。它采用Pose Neural Fabrics Search(PNFS)技术,有效提升了姿态估计的准确性和效率。
在人体结构的先验知识指导下搜索特定部位的基于细胞的神经结构(CNF)。这个存储库是我们的PyTorch实现论文的一部分。
安装依赖项:
通过pip install -r requirements.txt 安装 PyTorch (>=1.0.0) 和其他需要的包。
数据准备:我们按照相关步骤准备MPII和COCO数据集,请参考相应文档。
下载预训练模型:
- ImageNet 预训练 Resnet-50 (23.5M)
- MobileNet-V2(1.3M)
- HRNet-W32-stem~stage3 (8.1M)
创建输出目录以保存每个实验的输出:`mkdir o`
训练模型:
使用命令行执行 `python train.py --cfg configs/example.yaml` 来启动训练过程。
全部评论 (0)
还没有任何评论哟~


