SteveImmanuel / SegGPT-FineTune

Fine-tune SegGPT model with custom datasets

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Can I do multi-class segmentation with this repo??

sensationHJ opened this issue · comments

Hi! I'm really interested in this repo, because Seggpt's concept will be good for satellite image and, dataset of yours was satellite image also!

and I want to do segmentation on argriculture area, building, road(paved, dirt), water area.

can I make multi-class segmentation using this repo?

singe class segmentation will be good for preformence but using many of gpu will be risk for our system.

thanks!

Hi, sorry for the late reply.
I believe that the current code already supports segmentation for multiple class. Could you clarify your usage?