image-segmentation-keras
image-segmentation-keras copied to clipboard
How can I apply this model to grayscale images in medical image segmentation tasks?
trafficstars
I'm working on liver tumor segmentation using CT images which are grayscale images and have one channel. How can I apply my dataset to this model?
I have tried this way: (but doesn't work for SegNet here) inp = Input(shape=(480, 480, 1)) l1 = Conv2D(3, (1, 1))(inp) base_model = vgg_segnet(n_classes=2, input_height=480, input_width=480) out = base_model(l1) model = Model(inp, out, name=base_model.name) (learned from: Training with non RGB data https://segmentation-models.readthedocs.io/en/latest/tutorial.html#training-with-non-rgb-data)