Unverified Commit 70b13e1c authored by Pavel Yakubovskiy's avatar Pavel Yakubovskiy Committed by GitHub

Fix `prerprocess_input` for mobilenets (#56) (#98)

* Fix `prerprocess_input` for mobilenets (#56)

* Fix `preprocess_input` for mobilenets (#56)
parent 75075da9
......@@ -69,6 +69,7 @@ def preprocess_input(x, **kwargs):
# Returns
Preprocessed array.
"""
kwargs['backend'] = backend
return imagenet_utils.preprocess_input(x, mode='tf', **kwargs)
......
......@@ -89,6 +89,7 @@ def preprocess_input(x, **kwargs):
# Returns
Preprocessed array.
"""
kwargs['backend'] = backend
return imagenet_utils.preprocess_input(x, mode='tf', **kwargs)
......@@ -467,4 +468,4 @@ def _inverted_res_block(inputs, expansion, stride, alpha, filters, block_id):
if in_channels == pointwise_filters and stride == 1:
return layers.Add(name=prefix + 'add')([inputs, x])
return x
\ No newline at end of file
return x
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