Commit 17821c0d authored by syiming's avatar syiming
Browse files

fix doc string for get_box_classifier_feature_extractor_model

parent bd27aa70
...@@ -286,7 +286,7 @@ class FasterRCNNResnetV1FpnKerasFeatureExtractor( ...@@ -286,7 +286,7 @@ class FasterRCNNResnetV1FpnKerasFeatureExtractor(
And returns proposal_classifier_features: And returns proposal_classifier_features:
A 4-D float tensor with shape A 4-D float tensor with shape
[batch_size * self.max_num_proposals, 1024] [batch_size * self.max_num_proposals, 1, 1, 1024]
representing box classifier features for each proposal. representing box classifier features for each proposal.
""" """
with tf.name_scope(name): with tf.name_scope(name):
......
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