search

qubvel / segmentation_models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.
MIT License
4.77k stars 1.03k forks source link

Custom input shape for 3d model #507

Closed miladsade96 closed 2 years ago

miladsade96 commented 2 years ago

Hello, My data input shapes are (128, 128, 10, 3) and output shape is (128, 128, 10, 4). How can i handle this shapes in UNet model?