fix traveller59#9.

jdc08161063 · Oct 17, 2018 · 57af33a · 57af33a
1 parent 57a83cf
commit 57af33a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/second/core/target_ops.py b/second/core/target_ops.py
@@ -152,7 +152,7 @@ def create_target_np(all_anchors,
             labels[enable_inds] = 0
         bg_inds = np.where(labels == 0)[0]
     else:
-        if len(gt_boxes) == 0:
+        if len(gt_boxes) == 0 or anchors.shape[0] == 0:
             labels[:] = 0
         else:
             labels[bg_inds] = 0