med-air / 3DSAM-adapter

Holistic Adaptation of SAM from 2D to 3D for Promptable Medical Image Segmentation
134 stars 12 forks source link

Can I train my own dataset? #8

Open nanshanvv opened 1 year ago

nanshanvv commented 1 year ago

If I want to train my own dataset, how do I do it? Looking forward to your reply!

peterant330 commented 1 year ago

Hi,

I would recommend you download our arranged dataset to see how the data is organized and then arrange your own data accordingly. Then you can modify the data loader in 3DSAM-adapter/dataset/datasets.py according to your own data format.

brandonsct commented 8 months ago

If I want to train my own dataset, how do I do it? Looking forward to your reply!

Hi any progress on using other dataset? it would be great to see how it is done.