search

yuyinzhou / L2B

This repository includes the official project of L2B, from our paper "Learning to Bootstrap for Combating Label Noise".
28 stars 3 forks source link

L2B_Segmnetation missing ''from Models.Unetplpl_2D import NestedUNet'' #5

Open Kedaya66 opened 3 days ago

Kedaya66 commented 3 days ago

Dear author: Thanks for your contributions to noisy label learning. Now, I am trying to reimplement your work on segmentation task. However, it seems that the ''Models.Unetplpl_2D import NestedUNet'' isn't provided. Could you please upload this python file for better reimplementation. Thanks again!

Kedaya66 commented 3 days ago

after running script ''data_preprocess.py'', the ''meta_train_scan3 directory is empty. This affects the running of ''train.py'' script and has the following bug existed. ![Uploading image.png…]()

Kedaya66 commented 3 days ago

dataset: Prostate Norm_type org Img size: [144, 144] Experiments is conducted on: baseline_Unet++_2d_ema ==> Preparing data.. Traceback (most recent call last): File "E:\Anaconda\envs\dl\lib\site-packages\PIL\Image.py", line 3089, in fromarray mode, rawmode = _fromarray_typemap[typekey] KeyError: ((1, 1, 3, 512), '|u1')

The above exception was the direct cause of the following exception:

Traceback (most recent call last): File "D:\works\mission\Exploring\Z-CodesPool\L2B-main\L2B_segmentation\train.py", line 344, in main(args) File "D:\works\mission\Exploring\Z-CodesPool\L2B-main\L2B_segmentation\train.py", line 147, in main val_data, val_labels = next(iter(metaloader)) File "E:\Anaconda\envs\dl\lib\site-packages\torch\utils\data\dataloader.py", line 630, in next data = self._next_data() File "E:\Anaconda\envs\dl\lib\site-packages\torch\utils\data\dataloader.py", line 674, in _next_data data = self._dataset_fetcher.fetch(index) # may raise StopIteration File "E:\Anaconda\envs\dl\lib\site-packages\torch\utils\data_utils\fetch.py", line 51, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "E:\Anaconda\envs\dl\lib\site-packages\torch\utils\data_utils\fetch.py", line 51, in data = [self.dataset[idx] for idx in possibly_batched_index] File "D:\works\mission\Exploring\Z-CodesPool\L2B-main\L2B_segmentation\dataloader.py", line 176, in getitem img = Image.fromarray(np.expand_dims(img, 2).repeat(3, 2)) File "E:\Anaconda\envs\dl\lib\site-packages\PIL\Image.py", line 3092, in fromarray raise TypeError(msg) from e TypeError: Cannot handle this data type: (1, 1, 3, 512), |u1