wenxi-yue / SurgicalSAM

[AAAI2024] Official implementation of SurgicalSAM
MIT License
58 stars 7 forks source link

About the pre-process of val set #17

Open 1170300714 opened 4 weeks ago

1170300714 commented 4 weeks ago

Thanks for your great work!

I am referring to your preprocessing code to handle my private dataset. However, it seems that the data_preprocess.py only implements preprocessing for the training set. Could you please release the preprocessing code for the test set? Thank you!

wenxi-yue commented 3 weeks ago

Hi,

Thank you for your interest.

The preprocessing steps for the training set and the validation set are quite similar. The only difference is that for the validation set, we do not apply data augmentation. For processing the validation set, we use the original frames and original binary annotation masks (rather than the frames and binary annotation masks after augmentation) to compute the SAM features and class embeddings.

So you can refer to the code for computing the SAM features here, and the code on computing class embeddings here. In other words, you can use data_preprocess.py, but remove all parts related to data augmentation for processing the validation set.