lacss.losses

self_supervised_instance_loss(batch, prediction, *, soft_label=True)

Unsupervised instance loss

supervised_instance_loss(batch, prediction)

LACSS instance loss, supervised with segmentation label

weakly_supervised_instance_loss(batch, prediction, *, ignore_mask=False)

Instance loss supervised by image mask instead of instance masks

detection_loss(batch, prediction, *, gamma=2.0)

LPN detection loss

localization_loss(batch, prediction, *, delta=1.0)

LPN localization loss

lpn_loss(batch, prediction, *, gamma=2.0, w1=1.0, w2=1.0)

LPN loss

aux_size_loss(batch, prediction, *, weight=0.01)

Auxillary loss to prevent model collapse

self_supervised_edge_loss(batch, prediction)

Cell border prediction consistency loss

self_supervised_segmentation_loss(batch, prediction, *, offset_sigma=jnp.array([10.0]), offset_scale=jnp.array([2.0]))

Image segmentation consistenct loss for the collaboraor model