Skip to content

Commit

Permalink
Fix PyramidROIAlign output shape
Browse files Browse the repository at this point in the history
As discussed in: matterport#919
  • Loading branch information
waleedka committed Sep 28, 2018
1 parent c68f8a8 commit 6ece2f8
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion mrcnn/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -442,7 +442,8 @@ def call(self, inputs):
pooled = tf.gather(pooled, ix)

# Re-add the batch dimension
pooled = tf.expand_dims(pooled, 0)
shape = tf.concat([tf.shape(boxes)[:2], tf.shape(pooled)[1:]], axis=0)
pooled = tf.reshape(pooled, shape)
return pooled

def compute_output_shape(self, input_shape):
Expand Down

0 comments on commit 6ece2f8

Please sign in to comment.