Add Hugging Face integration

[NielsRogge]

Hi @yformer and team!

Thanks for this nice work. I wrote a quick PoC to showcase that you can easily have integration so that you can automatically load the various EfficientSAM models using from_pretrained (and push them using push_to_hub), track download numbers for your models (similar to models in the Transformers library), and have nice model cards on a per-model basis. It leverages the PyTorchModelHubMixin class which allows to inherits these methods.

Usage is as follows:

from efficient_sam import EfficientSAM

model = EfficientSAM.from_pretrained("nielsr/efficientsam-tiny")

The corresponding model is here for now: https://huggingface.co/nielsr/efficientsam-tiny. We could move all checkpoints to separate repos on your account on https://hf.co if you're interested.

This will make sure you can track download numbers for each individual checkpoint!

Would you be interested in this integration?

Kind regards,

Niels

[NielsRogge]

Hi @yformer any interest in having download metrics?

[yformer]

Hi @NielsRogge, thanks for integrating this. Yes, let me move all checkpoints to HF hub for tracking download numbers.

[yformer]

Hi @NielsRogge, I have uploaded the models for HF integration, https://huggingface.co/yunyangx/EfficientSAM. Do you want to update your change for HF integration? I will merge the pull request.

[NielsRogge]

Hi @yformer thanks for uploading the models. I'll push the models now!