huggingface / transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
https://huggingface.co/transformers
Apache License 2.0
132.06k stars 26.31k forks source link

Add RobustSAM (CVPR 2024) #32874

Open leolu030066 opened 4 weeks ago

leolu030066 commented 4 weeks ago

Model description

Robust Segment Anything Model (RobustSAM) is an enhanced version of the Segment Anything Model (SAM), designed to improve segmentation performance on low-quality images while preserving the original model's zero-shot segmentation capabilities and flexible prompting system.

Open source status

Provide useful links for the implementation

@robustsam

jadechoghari commented 3 weeks ago

@leolu030066 - Hello! RobustSAM is just an extension of SAM. We have implemented the HF model cards here: 👀🤗👇: https://github.com/robustsam/RobustSAM/issues/10#issuecomment-2294327808

Best.

leolu030066 commented 3 weeks ago

@leolu030066 - Hello! RobustSAM is just an extension of SAM. We have implemented the HF model cards here: 👀🤗👇: robustsam/RobustSAM#10 (comment)

Best.

Got it! Thanks for your contribution. Sincerely, Leo