bermanmaxim/LovaszSoftmax
A loss function layer implementing the Lovász-Softmax surrogate for optimizing IoU in neural network training.

Velocity · 7d
+0.5
★ / day
Trend
→steady
star history
The repository provides PyTorch and TensorFlow implementations of the Lovász-Softmax loss, a differentiable surrogate for the intersection-over-union metric used in image segmentation. It includes binary and multiclass variants suitable for training neural networks on semantic segmentation tasks. Published at CVPR 2018.