Fix mobilenet weight preprocessing.

Esse commit está contido em:
Francois Chollet
2017-06-30 13:52:14 -07:00
commit b42f6760bc
+7 -1
Ver Arquivo
@@ -72,7 +72,6 @@ from ..utils.data_utils import get_file
from ..engine.topology import get_source_inputs
from ..engine import InputSpec
from ..applications.imagenet_utils import _obtain_input_shape
from ..applications.imagenet_utils import preprocess_input
from ..applications.imagenet_utils import decode_predictions
from .. import backend as K
@@ -84,6 +83,13 @@ def relu6(x):
return K.relu(x, max_value=6)
def preprocess_input(x):
x /= 255.
x -= 0.5
x *= 2.
return x
class DepthwiseConv2D(Conv2D):
"""Depthwise separable 2D convolution.