lacss.losses

self_supervised_instance_loss(batch, prediction)

Unsupervised instance loss

supervised_instance_loss(batch, prediction)

LACSS instance loss, supervised with segmentation label

weakly_supervised_instance_loss(batch, prediction)

Instance loss supervised by image mask instead of instance masks

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

Auxillary loss to prevent model collapse

cks_boundry_loss(batch, prediction)

Cell border prediction consistency loss

cks_segmentation_loss(batch, prediction, *, offset_sigma=10.0, offset_scale=2.0)

Image segmentation consistenct loss for the collaboraor model