search

SysCV / sam-hq

Segment Anything in High Quality [NeurIPS 2023]
https://arxiv.org/abs/2306.01567
Apache License 2.0
3.73k stars 224 forks source link

I wanted to train the model with my images #57

Closed sumaoge closed 1 year ago

sumaoge commented 1 year ago

I wanted to train the model with my images. What should I do? Is it necessary to make my images to DIS5K dataset? What software can be used to create DIS5K file/dataset?

lkeab commented 1 year ago

Hi, you need to convert your images and masks annotation into the similar format provided in HQ-Seg-44K. You can download HQ-Seg-44K here. For dis, they use GIMP as their annotation tool but they are also other tools available as long as it can label masks.

sumaoge commented 1 year ago

Hi, you need to convert your images and masks annotation into the similar format provided in HQ-Seg-44K. You can download HQ-Seg-44K here. For dis, they use GIMP as their annotation tool but they are also other tools available as long as it can label masks.

THANKS!! I ' ll try

crapthings commented 1 year ago

@lkeab is mask pair okay? unzip trainning data is like this

image

i want to finetune a model, is it just load exist ckpt and feed new image mask pair?