minor fix

broadinstitute · May 31, 2017 · 2366dbb · 2366dbb
1 parent f15ccfb
commit 2366dbb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/keras_rcnn/backend/tensorflow_backend.py b/keras_rcnn/backend/tensorflow_backend.py
@@ -95,7 +95,7 @@ def overlap(x, y):
 
     k = y.shape[0]
 
-    overlaps = numpy.zeros((n, k), dtype=numpy.float32)
+    overlaps = keras.backend.zeros((n, k), dtype=numpy.float32)
 
     for k_index in range(k):
         area = ((y[k_index, 2] - y[k_index, 0] + 1) * (y[k_index, 3] - y[k_index, 1] + 1))

diff --git a/keras_rcnn/layers/object_detection/_object_proposal.py b/keras_rcnn/layers/object_detection/_object_proposal.py
@@ -9,7 +9,7 @@ class ObjectProposal(keras.engine.topology.Layer):
     """
 
     def __init__(self, proposals, **kwargs):
-        self.output_dim = (None, None, 4)
+        self.output_dim = (None, proposals, 4)
 
         self.proposals = proposals
 

diff --git a/tests/layers/object_detection/test_object_proposal.py b/tests/layers/object_detection/test_object_proposal.py
@@ -16,4 +16,4 @@ def test_call(self, object_proposal_model):
         assert prediction.shape == (1, 100, 4)
 
     def test_compute_output_shape(self, object_proposal_layer):
-        assert object_proposal_layer.compute_output_shape((14, 14)) == (None, None, 4)
+        assert object_proposal_layer.compute_output_shape((14, 14)) == (None, 300, 4)