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
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