Debug dsm.py

BMCV · Jul 16, 2024 · 57a7536 · 57a7536
1 parent 5d251b2
commit 57a7536
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/superdsm/dsm.py b/superdsm/dsm.py
@@ -337,8 +337,9 @@ def grad(self, params):
         grad = np.sum([term1 * q for q in self.q], axis=1)
         term1_sparse = coo_matrix(term1).transpose(copy=False)
         if self.smooth_mat.shape[1] > 0:
-            grad2  = (np.ones((1, np.prod(self.w.shape))) @ self.smooth_mat.multiply(term1_sparse)).reshape(-1)
+            #grad2  = (np.ones((1, np.prod(self.w.shape))) @ self.smooth_mat.multiply(term1_sparse)).reshape(-1)
             #grad2  = self.smooth_mat.multiply(term1_sparse).sum(axis=0).reshape(-1)
+            grad2  = self.smooth_mat.multiply(term1_sparse).sum(axis=0).reshape(-1)
             assert np.allclose(
                 (np.ones((1, np.prod(self.w.shape))) @ self.smooth_mat.multiply(term1_sparse)).reshape(-1),
                 self.smooth_mat.multiply(term1_sparse).sum(axis=0).reshape(-1)