PyTorch-PoseNet
5星
- 浏览量: 0
- 大小:None
- 文件类型：None
简介：
PyTorch-PoseNet是一款基于PyTorch框架的人体姿态估计工具包，适用于计算机视觉研究与应用开发，能够准确识别图像中人物的姿态关键点。
Pytorch-PoseNet 是一个根据ICCV 2015论文《PoseNet：实时六自由度摄像机重定位的卷积网络》实现的项目。该存储库中的PoseNet模型定义在PoseNet.py文件中，训练初始权重和经过训练后的权重（分别为posenet.npy和Posenet.ckpt）是通过转换张量流模型权重并进行训练后获得。
运行方法如下：
1. 将Cambridge Landmarks Kings College数据集提取到您喜欢的任何位置。
2. 将开始以及已经完成训练的权重文件提取到您希望的位置上。
3. 更新train.py中的路径（第12行）以指向正确的文件和目录。
4. 如果需要重新训练模型，只需运行train.py脚本。（注意：这将消耗大量时间。）
以上就是使用Pytorch-PoseNet的基本步骤说明。
全部评论 (0)
还没有任何评论哟~
