Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

【Hackathon + No.163】基于PaddleDetection PP-TinyPose,新增手势关键点检测模型 #8066

Merged
merged 5 commits into from
May 22, 2023

Conversation

flytocc
Copy link
Contributor

@flytocc flytocc commented Apr 11, 2023

训练

训练平台为AIStudio,PaddlePaddle2.4.0 with V100*4。训练脚本如下:

python -m paddle.distributed.launch --gpus=0,1,2,3 \
  tools/train.py \
  -c configs/keypoint/tiny_pose/tinypose_256x256_hand.yml \
  --eval
Models Size Epoch PCK@0.2 AUC EPE weight & log
TinyPose 256x256 100 0.7409 0.8037 5.6584 model_final.pdparams & train.log

权重及训练日志下载地址:百度网盘

@paddle-bot
Copy link

paddle-bot bot commented Apr 11, 2023

Thanks for your contribution!

@zhiboniu
Copy link
Collaborator

你好,感谢贡献。可以尝试把模型精度再提高一些,对比看LiteHRNet-18版本的精度PCK应该在0.795左右。

LiteHRNet-18 | 256x256 | 0.795 | 0.830 | 4.77

@flytocc
Copy link
Contributor Author

flytocc commented Apr 22, 2023

@zhiboniu 这是改进后的效果,精度与LiteHRNet-18相当接近。

Models Size Epoch AugmentationbyInformantionDropping PCK@0.2 AUC EPE weight & log
TinyPose 256x256 210 w/o 0.7951 0.8298 4.749 model_final_v2.pdparams & train_v2.log
TinyPose 256x256 420 w/o 0.7953 0.8311 4.725
TinyPose 256x256 420 w/ (prob 0.5 & factor 0.05) 0.7957 0.8307 4.717
TinyPose 256x256 420 w/ (prob 0.25 & factor 0.025) 0.7967 0.8321 4.681

权重及训练日志下载地址:百度网盘

zhiboniu
zhiboniu previously approved these changes Apr 24, 2023
@flytocc flytocc force-pushed the coco_whole_body_hand branch from d303e83 to 294a255 Compare May 9, 2023 08:27
@flytocc flytocc force-pushed the coco_whole_body_hand branch from cb78727 to 8259ff8 Compare May 22, 2023 04:48
@jerrywgz jerrywgz merged commit a694be1 into PaddlePaddle:develop May 22, 2023
@flytocc flytocc deleted the coco_whole_body_hand branch May 22, 2023 05:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants