I can't find a decent pytorch implementation of softmax focal loss for Semantic Segmentation. So I code one on my own.
type python loss.py to test the loss.
-
Notifications
You must be signed in to change notification settings - Fork 0
An intuitive pytorch implementation of softmax focal loss which can be used to train classification and segmentation tasks.
License
bot66/SoftmaxFocalLoss
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
About
An intuitive pytorch implementation of softmax focal loss which can be used to train classification and segmentation tasks.
Topics
Resources
License
Stars
Watchers
Forks
Releases
No releases published