avoid zero division in bbox2loc

chainer · Oct 20, 2017 · 3755023 · 3755023
1 parent 3802415
commit 3755023
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/chainercv/links/model/faster_rcnn/utils/bbox2loc.py b/chainercv/links/model/faster_rcnn/utils/bbox2loc.py
@@ -54,6 +54,9 @@ def bbox2loc(src_bbox, dst_bbox):
     base_ctr_y = dst_bbox[:, 0] + 0.5 * base_height
     base_ctr_x = dst_bbox[:, 1] + 0.5 * base_width
 
+    height = height + 1e-14
+    width = width + 1e-14
+
     dy = (base_ctr_y - ctr_y) / height
     dx = (base_ctr_x - ctr_x) / width
     dh = xp.log(base_height / height)