From 8dee552541cf0ad53309eb34bfa0cbc33bf02c3f Mon Sep 17 00:00:00 2001 From: HastingsGreer Date: Sat, 3 Feb 2024 11:33:29 -0500 Subject: [PATCH] Update losses.py --- src/icon_registration/losses.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/icon_registration/losses.py b/src/icon_registration/losses.py index 7226ea7..ba198f4 100644 --- a/src/icon_registration/losses.py +++ b/src/icon_registration/losses.py @@ -592,8 +592,8 @@ def __call__(self, image_A, image_B): 1 - (self.blur(I * J) - (self.blur(I) * self.blur(J))) / torch.sqrt( - (self.blur(I * I) - self.blur(I) ** 2 + 0.00001) - * (self.blur(J * J) - self.blur(J) ** 2 + 0.00001) + (torch.relu(self.blur(I * I) - self.blur(I) ** 2) + 0.00001) + * (torch.relu(self.blur(J * J) - self.blur(J) ** 2) + 0.00001) ) )