Skip to content

Commit

Permalink
Fix comments in Detection Layer
Browse files Browse the repository at this point in the history
1. fix description on window
2. fix output shape of detection layer
  • Loading branch information
keineahnung2345 authored and waleedka committed Sep 28, 2018
1 parent 6ece2f8 commit 30c652f
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions mrcnn/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -694,10 +694,10 @@ def refine_detections_graph(rois, probs, deltas, window, config):
probs: [N, num_classes]. Class probabilities.
deltas: [N, num_classes, (dy, dx, log(dh), log(dw))]. Class-specific
bounding box deltas.
window: (y1, x1, y2, x2) in image coordinates. The part of the image
window: (y1, x1, y2, x2) in normalized coordinates. The part of the image
that contains the image excluding the padding.
Returns detections shaped: [N, (y1, x1, y2, x2, class_id, score)] where
Returns detections shaped: [num_detections, (y1, x1, y2, x2, class_id, score)] where
coordinates are normalized.
"""
# Class IDs per ROI
Expand Down Expand Up @@ -817,7 +817,7 @@ def call(self, inputs):
self.config.IMAGES_PER_GPU)

# Reshape output
# [batch, num_detections, (y1, x1, y2, x2, class_score)] in
# [batch, num_detections, (y1, x1, y2, x2, class_id, class_score)] in
# normalized coordinates
return tf.reshape(
detections_batch,
Expand Down

0 comments on commit 30c652f

Please sign in to comment.