上传者: 42099633
|
上传时间: 2022-03-01 16:37:20
|
文件大小: 28KB
|
文件类型: -
PyTorch中的PoseLSTM和PoseNet实现
这是基于代码开发的PoseLSTM和PoseNet的PyTorch实现。
先决条件
Linux
的Python 3.5.2
CPU或NVIDIA GPU + CUDA CuDNN
入门
安装
从安装PyTorch和依赖项
克隆此仓库:
git clone https://github.com/hazirbas/posenet-pytorch
cd posenet-pytorch
pip install -r requirements.txt
PoseNet培训/测试
在datasets /文件夹下下载Cambridge Landscape数据集(例如 )。
计算均值图像
python util/compute_image_mean.py --dataroot datasets/KingsCollege --height 256