czg1225 / SlimSAM

SlimSAM: 0.1% Data Makes Segment Anything Slim
Apache License 2.0
248 stars 14 forks source link

Nice Work ! Could you provide the configuration about the SA-1B dataset? #5

Open xuke225 opened 5 months ago

xuke225 commented 5 months ago

Hi, could you please describe in detail how the SA-1B dataset is split into the training and the validation set in this work?

czg1225 commented 5 months ago

You can find the SA-1B dataset from url{https://ai.meta.com/datasets/segment-anything-downloads/}. The SA-1B dataset is divided into multiple subsets, each subset has 11k images. You can download any of the subsets, and then use 10k images as a training set, 50 images as a validationset, and the rest as a testset. Our experimental results were completed under the above configuration, but if the amount of training data is further increased, the performance of the model will be further improved. Thanks!

czg1225 commented 5 months ago

By the way, the 10k training dataset and 50 val dataset are splited randomly from the 11k subset.

xuke225 commented 5 months ago

Thank you for your response. As you indicated, I only need to download one of the compressed packages in SA-1B (e.g., sa_000000.tar) to complete the training, validation, and testing procedures for this task. However, will the test results have a large variance (due to different choices)?

czg1225 commented 5 months ago

We have tried different choices. The test results will be similar (MIoU variance < 0.5%).

xuke225 commented 5 months ago

Got it, thank you for your patient reply!