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

class balanced loss? #34

Open
2533245542 opened this issue Dec 22, 2021 · 1 comment
Open

class balanced loss? #34

2533245542 opened this issue Dec 22, 2021 · 1 comment

Comments

@2533245542
Copy link

Ever considered implementing class balanced loss? I found your repo useful and would like to see it include more.

https://github.com/vandit15/Class-balanced-loss-pytorch

@CoinCheung
Copy link
Owner

Sorry for being so late. I was busy with other things when this issue is created. Frankly, I do not have such a plan to add it because the implementation you posted above is good and well documented.

I tried it in sementic segmentation task of cityscapes, but observed a bit drop of miou. The model I used is deeplabv3+, and I simply replace cross entropy with this loss without other changings. Maybe I should be more careful to tune the training parameters.

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

2 participants