MedicineToken / Medical-SAM-Adapter

Adapting Segment Anything Model for Medical Image Segmentation
GNU General Public License v3.0
1.01k stars 89 forks source link

Do I need to create a corresponding dataset class for the BTCV example dataset? Because I see only the ISIC class in the dataset.py code. #38

Closed nanshanvv closed 8 months ago

AmrinKareem commented 1 year ago

The utils.py has the dataloader for BTCV

nanshanvv commented 1 year ago

The utils.py has the dataloader for BTCV

Thanks for your reply, I have found the dataloader. But when I run the code in the example, it still gives the following error: image Browsing the code I found that nice_train_loader=get_decath_loader(args) and get_decath_loader is present. Can you please tell me if you encountered this problem and how you solved it. image

WuJunde commented 10 months ago

you need to set -dataset as decathlon in the running