tlpss/keypoint-detection

There are plans to release larger models？

hnsywangxin opened this issue · 2 comments

hnsywangxin commented

Thanks for your amazing work, It helps me a lot.
But I found the model is very simple, the size of ckpt is less than 1MB，There are plans to release larger models？if not, can you give me some other repo or advise for reference. I need the model has stronger fitting ability.
Thanks.

tlpss commented

@hnsywangxin happy to hear that you find it helpful!

Tthere are a number of different models (refered to as 'backbone') available, some of which are pretrained on Imagenet. You can find them here.

For example, you can use in the CLI --backbone_type MaxVitUnet to use a 17M pretrained MaxViT model.

You can also easily add your own if want to.

hnsywangxin commented

@hnsywangxin happy to hear that you find it helpful!

Tthere are a number of different models (refered to as 'backbone') available, some of which are pretrained on Imagenet. You can find them here.

For example, you can use in the CLI --backbone_type MaxVitUnet to use a 17M pretrained MaxViT model.

You can also easily add your own if want to.

Thank you for your replay, I have followed your guidance and tried it, The results are very good. Thank you very much!