Error in Segmentation List Label

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):
        im = fn)[-1]
        image = crop_image_from_gray_blk(im[:,:,0])
        #print(image.shape )
        return ImageSegment(torch.tensor(image).float())