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

* Fix `prerprocess_input` for mobilenets (#56)

* Fix `preprocess_input` for mobilenets (#56)

Author: qubvel

segmentation_models/backbones/mobilenet.py

@@ -69,6 +69,7 @@ def preprocess_input(x, **kwargs):
     # Returns
         Preprocessed array.
     """
+    kwargs['backend'] = backend
     return imagenet_utils.preprocess_input(x, mode='tf', **kwargs)
 
 

segmentation_models/backbones/mobilenetv2.py

@@ -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
+    return x