fix metric

msracver · Dec 26, 2017 · ae742e2 · ae742e2
1 parent db4ef25
commit ae742e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fpn/core/metric.py b/fpn/core/metric.py
@@ -70,7 +70,7 @@ def update(self, labels, preds):
         num_classes = pred.shape[-1]
         # selection of ground truth label is different from softmax or sigmoid classifier
         label = label.asnumpy().reshape(-1, ).astype('int32')
-        fg_inds = np.where(label == 1)[0]
+        fg_inds = np.where(label > 0)[0]
         bg_inds = np.where(label == 0)[0]
         self.sum_metric += fg_inds.shape[0]
         self.num_inst += (fg_inds.shape[0] + bg_inds.shape[0])