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

Would improved loss functions like focal loss further boost the performance? #75

Open
LUPIN11 opened this issue Sep 1, 2024 · 0 comments

Comments

@LUPIN11
Copy link

LUPIN11 commented Sep 1, 2024

Thank you for your great work!
I notice that Segmenter just uses unweighted ce loss for training. However, ADE20K is not a dataset with balanced classes. I wonder whether we can boost current performance by improved loss functions like focal loss to compete with MaskFormer-like architectures.
I tried to do this but get extermely bad performance.
Have the authors or others attempted the same? Discussion is welcome!
My code for focal loss/dice loss is copied from:
https://github.com/qubvel-org/segmentation_models.pytorch/tree/main/segmentation_models_pytorch/losses

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

No branches or pull requests

1 participant