ymli39 / DeepSEED-3D-ConvNets-for-Pulmonary-Nodule-Detection

DeepSEED: 3D Squeeze-and-Excitation Encoder-Decoder ConvNets for Pulmonary Nodule Detection
MIT License
109 stars 33 forks source link

Dataloader has inputs other than size 128. #8

Closed gauthamnandakumar closed 4 years ago

gauthamnandakumar commented 5 years ago

Hey, When I load my dataloader, the input coming in is with image size other than 128. image

And for executing your step1.py file, to save the mask I followed the following code : out_folder = 'E:\LUNA Dataset\mask\' for idx,i in enumerate(target_filenames): filename = i.split('\')[-1] print(idx,i) case_pixels, bw1, bw2, spacing, origin = step1_python(i) bw = bw13+bw24 bw = sitk.GetImageFromArray(np.uint8(bw)) bw.SetSpacing(spacing) bw.SetOrigin(origin) sitk.WriteImage(bw,out_folder+filename, True)

Am I right?

ymli39 commented 5 years ago

Is this because your mask or original image has size smaller than 128?