提交 30c652f3 编写于 作者: keineahnung2345's avatar keineahnung2345 提交者: Waleed

Fix comments in Detection Layer

1. fix description on window
2. fix output shape of detection layer
上级 6ece2f84
...@@ -694,10 +694,10 @@ def refine_detections_graph(rois, probs, deltas, window, config): ...@@ -694,10 +694,10 @@ def refine_detections_graph(rois, probs, deltas, window, config):
probs: [N, num_classes]. Class probabilities. probs: [N, num_classes]. Class probabilities.
deltas: [N, num_classes, (dy, dx, log(dh), log(dw))]. Class-specific deltas: [N, num_classes, (dy, dx, log(dh), log(dw))]. Class-specific
bounding box deltas. 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. 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. coordinates are normalized.
""" """
# Class IDs per ROI # Class IDs per ROI
...@@ -817,7 +817,7 @@ class DetectionLayer(KE.Layer): ...@@ -817,7 +817,7 @@ class DetectionLayer(KE.Layer):
self.config.IMAGES_PER_GPU) self.config.IMAGES_PER_GPU)
# Reshape output # 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 # normalized coordinates
return tf.reshape( return tf.reshape(
detections_batch, detections_batch,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册