fix a bug in metric

speedinghzl · Aug 28, 2018 · 89a49c2 · 89a49c2
1 parent 160c5b4
commit 89a49c2
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/deeplab/metric.py b/deeplab/metric.py
@@ -51,8 +51,7 @@ def jaccard(self):
         jaccard = 0.0
         jaccard_perclass = []
         for i in xrange(self.nclass):
-            if not self.M[i, i] == 0:
-                jaccard_perclass.append(self.M[i, i] / (np.sum(self.M[i, :]) + np.sum(self.M[:, i]) - self.M[i, i]))
+            jaccard_perclass.append(self.M[i, i] / (np.sum(self.M[i, :]) + np.sum(self.M[:, i]) - self.M[i, i]))
 
         return np.sum(jaccard_perclass)/len(jaccard_perclass), jaccard_perclass, self.M