Skip to content

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