intel / unet

U-Net Biomedical Image Segmentation
Apache License 2.0
303 stars 124 forks source link

Train error for any dataset asides Tumor #34

Open b3r-prog opened 2 years ago

b3r-prog commented 2 years ago

Hello

I am trying to train on the differrent task. While the braintumor Task01 training works, the others dont. I get this error below

File ".../medical-decathlon/2D/dataloader.py", line 204, in generate_batch_from_files img = img[:,:,:,0] # Just take FLAIR channel (channel 0) IndexError: too many indices for array: array is 3-dimensional, but 4 were indexed

Is there an argument to specify or has this issue been noticed before?

@ravi9 @tonyreina