search

qubvel / segmentation_models

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

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

Open romanvelichkin opened 8 months ago

romanvelichkin commented 8 months ago

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