train Unet model with gray scale

qubvel / segmentation_models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

MIT License

4.66k stars 1.03k forks source link

Open paulwu808 opened 4 months ago

paulwu808 commented 4 months ago

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