Kornia - Read the Docs
https://kornia.readthedocs.io/en/latestKornia. State-of-the-art and curated Computer Vision algorithms for AI. Kornia AI is on the mission to leverage and democratize the next generation of Computer Vision tools and Deep Learning libraries within the context of an Open Source community. >>> import kornia.geometry as K >>> registrator = K.ImageRegistrator('similarity') >>> model ...
kornia.losses.dice - Kornia
https://kornia.readthedocs.io/en/latest/_modules/kornia/losses/dice.htmlAccording to [1], we compute the Sørensen-Dice Coefficient as follows:.. math:: \text{Dice}(x, class) = \frac{2 |X| \cap |Y|}{|X| + |Y|} Where: - :math:`X` expects to be the scores of each class. - :math:`Y` expects to be the one-hot tensor with the class labels. the loss, is finally computed as:.. math:: \text{loss}(x, class) = 1 - \text{Dice}(x, class) Reference: [1] https://en.wikipedia ...