Add SAM2 models and AI Batch Mode - Githubissues

wkentaro / labelme

Image Polygonal Annotation with Python (polygon, rectangle, circle, line, point and image-level flag annotation).

https://labelme.io

Other

13.62k stars 3.42k forks source link

Add SAM2 models and AI Batch Mode #1493

Open jakep72 opened 2 months ago

jakep72 commented 2 months ago

Support for Segment Anything 2 tiny and large models. Models are currently located in the fork release (see below):

SAM2 Large Models https://github.com/jakep72/labelme/releases/download/SAM2/sam2_large.encoder.onnx https://github.com/jakep72/labelme/releases/download/SAM2/sam2_large.decoder.onnx

SAM2 Tiny Models https://github.com/jakep72/labelme/releases/download/SAM2/sam2_hiera_tiny.encoder.onnx https://github.com/jakep72/labelme/releases/download/SAM2/sam2_hiera_tiny.decoder.onnx

Implement "AI Batch" mode -- adds the ability to label multiple similar polygons without clicking on each object. Accept or reject each polygon using the label dialog pop up widget.