Use spatial_squeeze=False for ResNet feature extractors. (tensorflow#…

…1586) Fixes tensorflow#1585
envin3 · Jun 16, 2017 · e76190e · e76190e
1 parent daf4bf9
commit e76190e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/object_detection/models/faster_rcnn_resnet_v1_feature_extractor.py b/object_detection/models/faster_rcnn_resnet_v1_feature_extractor.py
@@ -122,6 +122,7 @@ def _extract_proposal_features(self, preprocessed_inputs, scope):
               is_training=False,
               global_pool=False,
               output_stride=self._first_stage_features_stride,
+              spatial_squeeze=False,
               scope=var_scope)
 
     handle = scope + '/%s/block3' % self._architecture