diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-20 13:32:27 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-20 13:41:20 -0700 |
commit | c367ba02acc1d292738e3213173acbc0fe04335e (patch) | |
tree | f1e2a4bd05465191a9b2a227b9ca06efd1c5abcc /tensorflow/python/keras | |
parent | 05fe0cf733977572a546bf1e6117d491ca1dc221 (diff) |
This CL adds a Keras-based mobilenet_v2 feature extractor for object detection models.
As part of this CL, we use the Keras mobilenet_v2 application's keyword argument layer injection API to allow the generated network to support the object detection hyperparameters.
PiperOrigin-RevId: 213872175
Diffstat (limited to 'tensorflow/python/keras')
-rw-r--r-- | tensorflow/python/keras/applications/__init__.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/python/keras/applications/__init__.py b/tensorflow/python/keras/applications/__init__.py index a8b6d55e41..c35cdb15a4 100644 --- a/tensorflow/python/keras/applications/__init__.py +++ b/tensorflow/python/keras/applications/__init__.py @@ -63,7 +63,8 @@ def keras_modules_injection(base_fun): def wrapper(*args, **kwargs): if hasattr(keras_applications, 'get_submodules_from_kwargs'): kwargs['backend'] = backend - kwargs['layers'] = layers + if 'layers' not in kwargs: + kwargs['layers'] = layers kwargs['models'] = models kwargs['utils'] = utils return base_fun(*args, **kwargs) |