where is batchgenerators?

LucasFidon / trustworthy-ai-fetal-brain-segmentation

Trustworthy AI method based on Dempster-Shafer theory - application to fetal brain 3D T2w MRI segmentation

BSD 3-Clause "New" or "Revised" License

23 stars 2 forks source link

where is batchgenerators? #5

Closed DemerzelWang closed 7 months ago

DemerzelWang commented 7 months ago

Hi, I noticed that in the default_data_augmentation.py there are lines like from batchgenerators import ...., HOWEVER I didn't notice the existence of this file, and it rises to error when I tried to fine-tune on the nnUNet with command :"nnUNet_train 3d_fullres nnUNetTrainerV2_DRO_IS 50 1 -c ", could you please tell me how to solve this problem? I am more than delighted to hear your response!!

LucasFidon commented 7 months ago

Hi, thanks for your interest in our work.

batchgenerators is a python dependency available on pypi (you need version 0.21.0 to use with the version of nnUNet I use) https://pypi.org/project/batchgenerators/0.21/

It is a dependencies of the version of nnUNet I have used to train the models, see: https://github.com/LucasFidon/trustworthy-ai-fetal-brain-segmentation/blob/38cf1fb77bc896d133941b8f5388ef18feb2cae7/docker/third-party/nnUNet/setup.py#L18

I hope it helps!

Lucas

DemerzelWang commented 7 months ago

Thanks a lot!! That helps a lot! I am so stupid hhhh