Diff-UNet icon indicating copy to clipboard operation
Diff-UNet copied to clipboard

Function arguments of get_loader_btcv in test.py

Open ToruHironaka opened this issue 1 year ago • 1 comments

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)

ToruHironaka avatar May 12 '23 14:05 ToruHironaka

I have fixed the error, thank you.

920232796 avatar May 12 '23 15:05 920232796