上传者: 42120997
|
上传时间: 2021-11-22 21:33:09
|
文件大小: 6KB
|
文件类型: -
Pytorch-PoseNet
该存储库是实现。
如ICCV 2015年论文《 PoseNet:实时六自由度摄像机重定位的卷积网络》所述, Alex Kendall,Matthew Grimes和Roberto Cipolla [ ]
用法
下载Cambridge Landmarks King's College数据集
PoseNet模型在PoseNet.py文件中定义
训练的初始权重和训练权重(分别为posenet.npy和PoseNet.ckpt)是通过转换张量流模型权重然后进行训练而获得的。
跑步:
将国王学院的数据集提取到您喜欢的任何地方
将开始和训练过的权重提取到您喜欢的任何位置
更新第12行上的路径(train.py)
如果您想重新训练,只需运行train.py(请注意,这将花费很长时间)