Use spatial_squeeze=False for ResNet feature extractors. (#1586)

Fixes #1585

Use spatial_squeeze=False for ResNet feature extractors. (#1586)
Fixes #1585
e76190e8 · derekjchow · Sergio Guadarrama · daf4bf93 · e76190e8
Commit e76190e8 authored Jun 16, 2017 by derekjchow Committed by Sergio Guadarrama Jun 16, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

object_detection/models/faster_rcnn_resnet_v1_feature_extractor.py ...tection/models/faster_rcnn_resnet_v1_feature_extractor.py +1 -0

No files found.
--- 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 @@ class FasterRCNNResnetV1FeatureExtractor(
              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