qubvel / segmentation_models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Is there option to add classification head after encoder like in pytorch version?

romanvelichkin opened this issue · comments

In pytorch version of "segmentation models" it really easy to do - with high level API. But what about this one made for Keras?