Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Pretrained Weights

Latest

Choose a tag to compare

@alanqrwang alanqrwang released this 07 Jun 14:10
· 263 commits to main since this release
5b57d86

Pretrained weights used in the paper.

Naming convention for trained models:
numkey<number of keypoints>_<affine or tps>_<loss_fn>.<epochs>.h5
dice is supervised, mse is unsupervised, msekpconsistency is unsupervised with keypoint consistency loss.

We also provide the self-supervised pretrained models.
Naming convention for pretrained models:
numkey<number of keypoints>_pretrain.<epochs>.h5