jrmathew1

UNET 3plus not learning on large datasets

1

I am trying to use the UNET 3plus with deep supervision and cgm for nodule segmentation from lung ct scans. When I train the model on a 1000 image subset...

I am training the model on a new dataset and got this error during epoch 6: File "/data2/NoduleNet/dataset/mask_reader.py", line 124, in __getitem__ return [torch.from_numpy(input).float(), truth_bboxes, truth_labels, truth_masks, masks] UnboundLocalError: local...