qubvel / segmentation_models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

immuhammadadil opened this issue · comments

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.