search

ByungKwanLee / Full-Segment-Anything

This is Pytorch Implementation Code for adding new features in code of Segment-Anything. Here, the features support batch-input on the full-grid prompt (automatic mask generation) with post-processing: removing duplicated or small regions and holes, under flexible input image size
MIT License
134 stars 9 forks source link

Prompt flexibility #2

Open ramdhan1989 opened 8 months ago

ramdhan1989 commented 8 months ago

Hi, can I use custom point prompts instead of grid? and is it possible to not limit the number of points?

ByungKwanLee commented 8 months ago

Sorry for the late reply.

Yes sure! you can flexibly custom point prompts instead of grids and you can see the official code document or notebook example in https://github.com/facebookresearch/segment-anything/blob/main/notebooks/predictor_example.ipynb

My code is compatiable with it!

ramdhan1989 commented 8 months ago

Hi, do you have idea to improve my case?

image

blue and red dot is my prompts. Actually, I need to assign color for each layer. I think, SAM prompt can be used for background and foreground only?

ByungKwanLee commented 8 months ago

I think segmenting line-like thing having thin width can be conducted using High-Quality SAM [link]