search

tlpss / keypoint-detection

2D keypoint detection with Pytorch Lightning and wandb
MIT License
70 stars 9 forks source link

There are plans to release larger models？ #37

Closed hnsywangxin closed 11 months ago

hnsywangxin commented 11 months ago

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 11 months ago

@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 11 months ago

@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!