search

yformer / EfficientSAM

EfficientSAM: Leveraged Masked Image Pretraining for Efficient Segment Anything
Apache License 2.0
1.97k stars 142 forks source link

Integration to segment_anything package? #19

Open oranshayer opened 6 months ago

oranshayer commented 6 months ago

Hey,

Thanks for your work and model! Is there a plan to integrate this model to work with the segment_anything package, specifically with SamAutomaticMaskGenerator?

For example, this is how we use it today:

        self.sam = sam_model_registry["vit_h"](checkpoint=model_checkpoint_path).to(self.device)
        self.mask_generator = SamAutomaticMaskGenerator(self.sam, pred_iou_thresh=0.88, stability_score_thresh=0.8, min_mask_region_area=200)

will we be able to use the same with the EfficiantSAM model? Thanks!

yformer commented 6 months ago

Hi @oranshayer, thanks for your interest! Yes, we are planning EfficientSAM with the segment_anything package. We will let you know when it is ready.

AmitIsraeli commented 6 months ago

thank you very much waiting for it also (:

smandava98 commented 6 months ago

following

hadariru commented 2 weeks ago

following