Fix assertion of pairwise potential shape, thx @Pandoro.

b5e13a9a · lucasb-eyer · c15c0e55 · b5e13a9a
Commit b5e13a9a authored Apr 25, 2017 by lucasb-eyer
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

pydensecrf/densecrf.pyx pydensecrf/densecrf.pyx +2 -2

No files found.
--- a/pydensecrf/densecrf.pyx
+++ b/pydensecrf/densecrf.pyx
@@ -69,8 +69,8 @@ cdef class DenseCRF:
            del self._this
    def addPairwiseEnergy(self, float[:,::1] features not None, compat, KernelType kernel=DIAG_KERNEL, NormalizationType normalization=NORMALIZE_SYMMETRIC):
-        if features.shape[0] != self._nlabel or features.shape[1] != self._nvar:
+        if features.shape[1] != self._nvar:
-            raise ValueError("Bad shape for pairwise energy (Need {}, got {})".format((self._nlabel, self._nvar), (features.shape[0], features.shape[1])))
+            raise ValueError("Bad shape for pairwise energy (Need (?, {}), got {})".format(self._nvar, (features.shape[0], features.shape[1])))
        self._this.addPairwiseEnergy(eigen.c_matrixXf(features), _labelcomp(compat), kernel, normalization)