stillleben.losses module

Loss functions to be used with stillleben differentiation package.

Contents

Functions

def neg_iou_loss(predict, target)
Negative intersection over union loss function.

Function documentation

def stillleben.losses.neg_iou_loss(predict, target)

Negative intersection over union loss function.

Parameters
predict BxCxHxW image
target BxCxHxW image
Returns B vector of loss values, loss visualization