The Wayback Machine - https://web.archive.org/web/20210328061742/https://github.com/topics/image-processing
Skip to content
#

Image processing

Digital image processing is the use of algorithms to make computers analyze the content of digital images.

Here are 9,425 public repositories matching this topic...

albumentations
marlinsiebert
marlinsiebert commented Mar 19, 2021

🐛 Bug

When computing the binary focal loss I got nan gradients on backward pass when torch.sig(prediction) equals 1. or 0. and 0.<gamma<1..

This probably does also concern the focal loss implementation.

To Reproduce

import torch
from kornia.losses import binary_focal_loss_with_logits

prediction=torch.tensor([[-100.,-100.,100.,100.]], requires_grad=True)