← all repositories

bermanmaxim/LovaszSoftmax

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

1.4k stars Jupyter Notebook ML FrameworksComputer Vision
LovaszSoftmax
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.