Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

segmentation models - make num_classes and softmax non must #2476

Open
yonigottesman opened this issue Aug 16, 2024 · 1 comment
Open

segmentation models - make num_classes and softmax non must #2476

yonigottesman opened this issue Aug 16, 2024 · 1 comment
Assignees

Comments

@yonigottesman
Copy link

Currently all the segmentation models deep_lab_v3, segformer etc.. create the final layer with num_classes and always does softmax. Maybe we should have an activation parameter and if its None dont do any activation.
Some classification models require sigmoid and sometimes I want the unet for generative tasks without and activation

@mehtamansi29 mehtamansi29 added the enhancement New feature or request label Sep 17, 2024
@sachinprasadhs
Copy link
Collaborator

Thanks for reporting the issue.
We have the re enhanced DeepLab and other models in Keras Hub which has a option to provide the activation .
For more details check https://github.com/keras-team/keras-hub/blob/master/keras_hub/src/models/deeplab_v3/deeplab_v3_segmenter.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

3 participants