Skip to content

GitLab

Commit fd7e8be5 authored by syiming's avatar syiming
Browse files

concat all anchors

parent 1c6ba420
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
research/object_detection/meta_architectures/faster_rcnn_meta_arch.py
View file @ fd7e8be5
...@@ -1332,6 +1332,7 @@ class FasterRCNNMetaArch(model.DetectionModel): ...@@ -1332,6 +1332,7 @@ class FasterRCNNMetaArch(model.DetectionModel):
single_rpn_box_predictor_features = ( single_rpn_box_predictor_features = (
self._first_stage_box_predictor_first_conv(single_rpn_features_to_crop)) self._first_stage_box_predictor_first_conv(single_rpn_features_to_crop))
rpn_box_predictor_features.append(single_rpn_box_predictor_features) rpn_box_predictor_features.append(single_rpn_box_predictor_features)
anchors = tf.concat(anchors, axis=0)
return (rpn_box_predictor_features, rpn_features_to_crop, return (rpn_box_predictor_features, rpn_features_to_crop,
anchors, image_shape) anchors, image_shape)
......
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