Don't assume regression size.

huynhtnhut97 · Oct 9, 2018 · fcd6da4 · fcd6da4
1 parent 2d065c2
commit fcd6da4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/keras_retinanet/losses.py b/keras_retinanet/losses.py
@@ -90,8 +90,8 @@ def _smooth_l1(y_true, y_pred):
         """
         # separate target and state
         regression        = y_pred
-        regression_target = y_true[:, :, :4]
-        anchor_state      = y_true[:, :, 4]
+        regression_target = y_true[:, :, :-1]
+        anchor_state      = y_true[:, :, -1]
 
         # filter out "ignore" anchors
         indices           = backend.where(keras.backend.equal(anchor_state, 1))

diff --git a/keras_retinanet/utils/anchors.py b/keras_retinanet/utils/anchors.py
@@ -107,7 +107,7 @@ def anchor_targets_bbox(
             anchors_centers = np.vstack([(anchors[:, 0] + anchors[:, 2]) / 2, (anchors[:, 1] + anchors[:, 3]) / 2]).T
             indices = np.logical_or(anchors_centers[:, 0] >= image.shape[1], anchors_centers[:, 1] >= image.shape[0])
 
-            labels_batch[index, indices, -1]     = - 1
+            labels_batch[index, indices, -1]     = -1
             regression_batch[index, indices, -1] = -1
 
     return labels_batch, regression_batch