栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

如何训练LightWeight OpenPose

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

如何训练LightWeight OpenPose

前期准备:

1 下载COCO2017数据集,数据集地址为:http://cocodataset.org/#download](http://cocodataset.org/#download

解压至文件夹

2 安装要求的环境  pip install -r requirements.txt
训练

训练包含三个步骤

1.训练MobileNet权重,经过此轮训练期望AP值达到38%

2.训练上一阶段的权重,期待达到AP39%

3训练上一阶段的权重,期待达到AP40%

具体步骤:

1.下载MobileNet v1权重 mobilenet_sgd_68.848.pth.tar
2. 转换标签格式

python scripts/prepare_train_labels.py --labels /annotations/person_keypoints_train2017.json

产生prepared_train_annotation.pkl

3.python train.py --train-images-folder /train2017/ --prepared-train-labels prepared_train_annotation.pkl --val-labels val_subset.json --val-images-folder /val2017/ --checkpoint-path /mobilenet_sgd_68.848.pth.tar --from-mobilenet

4. python train.py --train-images-folder /train2017/ --prepared-train-labels prepared_train_annotation.pkl --val-labels val_subset.json --val-images-folder /val2017/ --checkpoint-path /checkpoint_iter_420000.pth --weights-only

5. 继续训练,三步调优, python train.py --train-images-folder /train2017/ --prepared-train-labels prepared_train_annotation.pkl --val-labels val_subset.json --val-images-folder /val2017/ --checkpoint-path /checkpoint_iter_280000.pth --weights-only --num-refinement-stages 3  采用 checkpoint 370000 迭代后作为最终的结果.

We did not perform the best checkpoint selection at any step, so similar result may be achieved after less number of iterations.

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/863797.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号