Hi i get the following error when i try to create Seg image databunch
expected 4D or 5D input and grid with same number of dimensions, but got input with sizes [1, 224, 224] and grid with sizes [1, 224, 224, 2]
def open(self, fn):
fn=str(MASK_PATH)+'/'+fn+'_mask.tiff'
im = skimage.io.MultiImage( fn)[-1]
image = crop_image_from_gray_blk(im[:,:,0])
image=cv2.resize(image,(224,224))
#print(image.shape )
return ImageSegment(torch.tensor(image).float())