From a673f807c5f48394ce28428ba5579b13cbb55f22 Mon Sep 17 00:00:00 2001 From: CptCaptain Date: Mon, 13 Dec 2021 17:34:53 +0100 Subject: [PATCH] Use pytorch photometric loss instead of old self-rolled --- model/networks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model/networks.py b/model/networks.py index a40c527..d6e06d9 100644 --- a/model/networks.py +++ b/model/networks.py @@ -387,7 +387,7 @@ class RectifiedPatternSimilarityLoss(TimedModule): if std is not None: mask = mask * std - diff = torchext.photometric_loss(pattern_proj.contiguous(), im.contiguous(), 9, self.loss_type, self.loss_eps) + diff = torchext.photometric_loss_pytorch(pattern_proj.contiguous(), im.contiguous(), 9, self.loss_type, self.loss_eps) val = (mask * diff).sum() / mask.sum() return val, pattern_proj