qubvel / segmentation_models

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

TypeError: Invalid shape (384, 480, 8) for image data #551

Closed immuhammadadil closed 1 year ago

immuhammadadil commented 1 year ago

When visualizing results for 8 class mask, following error occurred: TypeError: Invalid shape (384, 480, 8) for image data

I am running multiclass example for this repo.