HQ-SAM is now available in 🤗 Transformers

[NielsRogge]

Hi there,

The HQ-SAM model is officially integrated in the Hugging Face Transformers library 🤗 thanks to https://github.com/huggingface/transformers/pull/35147 (amazing work by @sushmanthreddy !)

It enables easy inference in a few lines of code as well as fine-tuning on custom data.

Resources

• Models can be found here: https://huggingface.co/syscv-community
• Docs: https://huggingface.co/docs/transformers/main/en/model_doc/sam_hq
• Inference notebook: https://github.com/NielsRogge/Transformers-Tutorials/tree/master/HQ-SAM

Fine-tuning

See my tutorial notebook on fine-tuning SAM or HQ-SAM on your custom dataset.

Cheers,

Niels and the open-source team at HF

[HamadYA]

Hi there,

The HQ-SAM model is officially integrated in the Hugging Face Transformers library 🤗 thanks to huggingface/transformers#35147 (amazing work by @sushmanthreddy !)

It enables easy inference in a few lines of code as well as fine-tuning on custom data.

Resources

• Models can be found here: https://huggingface.co/syscv-community
• Docs: https://huggingface.co/docs/transformers/main/en/model_doc/sam_hq
• Inference notebook: https://github.com/NielsRogge/Transformers-Tutorials/tree/master/HQ-SAM

Fine-tuning

See my tutorial notebook on fine-tuning SAM or HQ-SAM on your custom dataset.

Cheers,

Niels and the open-source team at HF

Greetings, Thanks for your amazing work.

Do you have access to the training code of HQ-SAM2? I tried to train it following the same steps as in HQ-SAM and it did not work for me.

If it is not available then it is fine.

Thanks again for your work.