Open emrullahpolat opened 4 months ago
Hi @emrullahpolat, unfortunately, it is not that trivial, but you can do it. Here is an example https://segmentation-models-pytorch.readthedocs.io/en/latest/insights.html#creating-your-own-encoder
Thanks @qubvel, Can I use the these codes Keras backend? It seems as Pytorch.
Not sure if you can, I don't have experience with Keras+(Pytorch backend). Let me know if you will be able to use it :)
Hello, I want to use the UNet of Segmentation Model, I tried VGG and effnetB3 to add backbone. I wonder if I can add my own Keras feature extraction model. For example ;
model = sm.Linknet(BACKBONE=feature_extract_model , classes=n_classes, activation=activation, encoder_weights=None, encoder_freeze=True)
Do you think It is true?