"tests/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "552cd32058660573c14ac481d88417e828c68756"
Commit a836069a authored by syiming's avatar syiming
Browse files

Fix predict second stage recive a list rpn_features

rpn_features for rfcn at second stage only have 1 level.
parent 503cc99c
...@@ -265,7 +265,7 @@ class RFCNMetaArch(faster_rcnn_meta_arch.FasterRCNNMetaArch): ...@@ -265,7 +265,7 @@ class RFCNMetaArch(faster_rcnn_meta_arch.FasterRCNNMetaArch):
[batch_size, num_valid_anchors, 2] containing class [batch_size, num_valid_anchors, 2] containing class
predictions (logits) for each of the anchors. Note that this predictions (logits) for each of the anchors. Note that this
tensor *includes* background class predictions (at class index 0). tensor *includes* background class predictions (at class index 0).
rpn_features: A 4-D float32 tensor with shape rpn_features: A list of single 4-D float32 tensor with shape
[batch_size, height, width, depth] representing image features from the [batch_size, height, width, depth] representing image features from the
RPN. RPN.
anchors: 2-D float tensor of shape anchors: 2-D float tensor of shape
...@@ -312,7 +312,8 @@ class RFCNMetaArch(faster_rcnn_meta_arch.FasterRCNNMetaArch): ...@@ -312,7 +312,8 @@ class RFCNMetaArch(faster_rcnn_meta_arch.FasterRCNNMetaArch):
_) = self._postprocess_rpn(rpn_box_encodings, _) = self._postprocess_rpn(rpn_box_encodings,
rpn_objectness_predictions_with_background, rpn_objectness_predictions_with_background,
anchors, image_shape_2d, true_image_shapes) anchors, image_shape_2d, true_image_shapes)
rpn_features = rpn_features[0]
box_classifier_features = ( box_classifier_features = (
self._extract_box_classifier_features(rpn_features)) self._extract_box_classifier_features(rpn_features))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment