python绘制图形沙漏_pytorch-pose一个用于二维人体姿势估计的PyTorch工具包。 - pytorch中文网...

pytorch-pose

PyTorch-Pose是2D单人姿态估计的一般流水线的PyTorch实现。其目的是为最流行的人体姿态数据库(如MPII人体姿态,LSP和FLIC)提供训练/推理/评估的接口,以及带各种数据增强选项的数据采集器。

用于数据准备和增强的一些代码来自Stacked沙漏网络。感谢原作者。

特征

多线程数据加载

多GPU训练

记录仪

培训/测试结果可视化

安装

1、PyTorch(>=0.2.0):请按照PyTorch的安装说明进行操作。请注意,代码是用Python2开发的,还没有用Python3进行过测试。

2、使用子模块克隆存储库

git clone --recursive https://github.com/bearpaw/pytorch-pose.git

创建一个符号链接到imagesMPII数据集的目录:

ln -s PATH_TO_MPII_IMAGES_DIR data/mpii/images

用法

测试

您可以下载我们的预训练模型(例如,双层沙漏模型)以便快速启动。

在终端运行以下命令以评估MPII验证分割模型(train / val split来自Tompson et al. CVPR 2015)。

CUDA_VISIBLE_DEVICES=0 python example/mpii.py -a hg --stacks 2 --blocks 1 --checkpoint checkpoint/mpii/hg_s2_b1 --resume checkpoint/mpii/hg_s2_b1/model_best.pth.tar -e -d

-a指定网络体系结构

--resume 将从一个特定的

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值