search

ge-xing / Diff-UNet

Diff-UNet: A Diffusion Embedded Network for Volumetric Segmentation. (using diffusion for 3D medical image segmentation)
Apache License 2.0
155 stars 21 forks source link

Function arguments of get_loader_btcv in test.py #15

Open ToruHironaka opened 1 year ago

ToruHironaka commented 1 year ago

BTCV test.py script's code below at line 151. The function get_loader_btcv's arguments set batch_size, but this function parameter does not have batch_size. Here is the function parameter get_loader_btcv(data_dir, cache=True). I assume data_dir=./RawData/Training because get_loader_btcv read image and label files and RawData/Testing has only img directory. Am I setting the function arguments correct? or Am I using a wrong dataset? Please let me know.

train_ds, val_ds, test_ds = get_loader_btcv(batch_size=batch_size, fold=0, cache=False)

920232796 commented 1 year ago

I have fixed the error, thank you.