Curt-Park / segment-anything-with-clip

Segment Anything combined with CLIP
https://huggingface.co/spaces/curt-park/segment-anything-with-clip
Apache License 2.0
328 stars 22 forks source link

Suggestion - Integrate MobileSAM into the pipeline for lightweight and faster inference #6

Open mdimtiazh opened 1 year ago

mdimtiazh commented 1 year ago

Reference: https://github.com/ChaoningZhang/MobileSAM

Our project performs on par with the original SAM and keeps exactly the same pipeline as the original SAM except for a change on the image encode, therefore, it is easy to Integrate into any project.

MobileSAM is around 60 times smaller and around 50 times faster than original SAM, and it is around 7 times smaller and around 5 times faster than the concurrent FastSAM. The comparison of the whole pipeline is summarzed as follows:

image

image

Best Wishes,

Qiao

kevin9foong commented 12 months ago

Sharing my observations on this: With mobileSAM, the inference improvements are not as good (only 2x improvement due to the lightweight image encoder) as mobileSAM is not optimised for everything mode which segment-anything-with-clip uses, using FastSAM might give faster inference but i have not tested this

This is due to the everything mode performing the decoder step n_points_per_side * n_points_per_side times