search

MrGiovanni / UNetPlusPlus

[IEEE TMI] Official Implementation for UNet++
Other
2.26k stars 538 forks source link

issue in resize_segmentation #79

Open mukul-B opened 2 years ago

mukul-B commented 2 years ago

File "/home/mukbad/anaconda3/envs/mypyt/lib/python3.9/site-packages/batchgenerators/dataloading/multi_threaded_augmenter.py", line 48, in producer item = transform(item) File "/home/mukbad/anaconda3/envs/mypyt/lib/python3.9/site-packages/batchgenerators/transforms/abstract_transforms.py", line 88, in call data_dict = t(data_dict) File "/mnt/d/PycharmProjects/UNetPlusPlus/nnunet/training/data_augmentation/downsampling.py", line 83, in call data_dict[self.output_key] = downsample_seg_for_ds_transform2(data_dict[self.input_key], self.ds_scales, self.order, File "/mnt/d/PycharmProjects/UNetPlusPlus/nnunet/training/data_augmentation/downsampling.py", line 103, in downsample_seg_for_ds_transform2 out_seg[b, c] = resize_segmentation(seg[b, c], new_shape[2:], order, cval) TypeError: resize_segmentation() takes from 2 to 3 positional arguments but 4 were given

mukul-B commented 2 years ago

Is anyone looking at this?

maryjis commented 2 years ago

Have the same issue

maryjis commented 2 years ago

You can resolve it by removing cval param, like this:

resize_segmentation(seg[b, c], new_shape[2:], order)