open-mmlab / mmpose

OpenMMLab Pose Estimation Toolbox and Benchmark.
https://mmpose.readthedocs.io/en/latest/
Apache License 2.0
5.4k stars 1.18k forks source link

transfer learning on pose models #1700

Closed adamsau closed 1 year ago

adamsau commented 1 year ago

is it possible to do transfer learning on HRNet / Lite HRNet? I am trying to do keypoint detection on objects, not human. Any guidance on that?

jin-s13 commented 1 year ago

MMPose will load imagenet-pretrained model for transfer learning. If you have enough data (>hundreds of imgs), you can train a good enough model for objects.

adamsau commented 1 year ago

https://mmpose.readthedocs.io/en/latest/papers/backbones.html#topdown-heatmap-litehrnet-on-coco @jin-s13 can you share the link of Lite HRNet weights trained on imagenet? I can't seem to find the pretrained weights of different models on imagenet.

jin-s13 commented 1 year ago

lite-HRNet does not have imagenet pre-trained models. But HRNet has.

adamsau commented 1 year ago

is there a link for all models pretrained on imagenet?

jin-s13 commented 1 year ago

Most models can be found in MMClassification and mmcv modelzoo.