train Unet model with gray scale

qubvel / segmentation_models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

MIT License

4.76k stars 1.03k forks source link

Open paulwu808 opened 8 months ago

paulwu808 commented 8 months ago

Hello! I wanna ask what code should i change if I wanna train a model with gray scale images

glbessa commented 4 months ago

Hello paulwu808, the module documentation has a section about it. You can find it here.