fix division by zero bug when calc iou

youtubevos · Jul 19, 2019 · fc8ec1d · fc8ec1d
1 parent a5c1f86
commit fc8ec1d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/PythonAPI/pycocotools/ytvoseval.py b/PythonAPI/pycocotools/ytvoseval.py
@@ -208,7 +208,8 @@ def iou_seq(d_seq, g_seq):
                     u += maskUtils.area(g)
                 elif d and not g:
                     u += maskUtils.area(d)
-            return i / u
+            iou = i / u if u > .0 else .0
+            return iou
         ious = np.zeros([len(d), len(g)])
         for i, j in np.ndindex(ious.shape):
             ious[i, j] = iou_seq(d[i], g[j])