Set regularize_depthwise to true in mobilenet_v1_argscope.

PiperOrigin-RevId: 192472546

Set regularize_depthwise to true in mobilenet_v1_argscope.
PiperOrigin-RevId: 192472546
0940134e · Zhichao Lu · pkulzc · b47ca971 · 0940134e
Commit 0940134e authored Apr 11, 2018 by Zhichao Lu Committed by pkulzc Apr 13, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

research/object_detection/models/ssd_mobilenet_v1_feature_extractor.py ...ct_detection/models/ssd_mobilenet_v1_feature_extractor.py +2 -1

No files found.
--- a/research/object_detection/models/ssd_mobilenet_v1_feature_extractor.py
+++ b/research/object_detection/models/ssd_mobilenet_v1_feature_extractor.py
@@ -105,7 +105,8 @@ class SSDMobileNetV1FeatureExtractor(ssd_meta_arch.SSDFeatureExtractor):
    with tf.variable_scope('MobilenetV1',
                           reuse=self._reuse_weights) as scope:
      with slim.arg_scope(
-          mobilenet_v1.mobilenet_v1_arg_scope(is_training=None)):
+          mobilenet_v1.mobilenet_v1_arg_scope(
+              is_training=None, regularize_depthwise=True)):
        with (slim.arg_scope(self._conv_hyperparams_fn())
              if self._override_base_feature_extractor_hyperparams
              else context_manager.IdentityContextManager()):