facebookresearch / facebookresearch/segment-anything

How to infer mask for multiple images?

Open
#492 4 comments 1 reaction 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
54.9k
Forks
6.4k
PR merge metrics
No merged PRs in 30d

Description

Thanks for your great work!

I would like to use the mask generator (see [notebooks/automatic_mask_generator_example.ipynb](https://github.com/facebookresearch/segment-anything/blob/main/notebooks/automatic_mask_generator_example.ipynb)) to get masks for multiple images. Therefore, the dimension will be 4 (**batch, channel, height, width**). It seems that only predictor can do that?

Contributor guide

Open the contributing guide

Research direction

Start with notebooks/automatic_mask_generator_example.ipynb and trace the mask generator and predictor entry points for their expected image dimensions. Determine whether multiple images with batch, channel, height, and width are supported; done means the repository has a clear answer or documented support path for batched inference.

Written by the indexing model from the issue text.

Assessment

Tech stack
jupyter-notebook
Domain
computer-vision, machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.