search

tianrun-chen / SAM-Adapter-PyTorch

Adapting Meta AI's Segment Anything to Downstream Tasks with Adapters and Prompts
MIT License
968 stars 83 forks source link

The structure of the dataset #10

Closed yyf17 closed 1 year ago

yyf17 commented 1 year ago

Very wonderful work! Can you provide the structure of the dataset?

tianrun-chen commented 1 year ago

The setup is straightforward - there is a folder for masks that contains the mask files, and another folder for images that stores the corresponding RGB images (it is important to verify the file names in the dataloader prior to training). The matching of masks and RGB images is based on their shared file names, which are usually numbered from 1 to N, where N represents the total number of images.