From 43ab5bf2b281d6ff17d4357bba149ec5c0a433d2 Mon Sep 17 00:00:00 2001 From: BbChip0103 Date: Tue, 24 Nov 2020 15:45:29 +0900 Subject: [PATCH] Update preprocess_input function for tf.data --- efficientnet/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/efficientnet/model.py b/efficientnet/model.py index 42371ff..8805a43 100644 --- a/efficientnet/model.py +++ b/efficientnet/model.py @@ -33,7 +33,7 @@ from six.moves import xrange from keras_applications.imagenet_utils import _obtain_input_shape -from keras_applications.imagenet_utils import preprocess_input as _preprocess_input +from tensorflow.keras.applications.imagenet_utils import preprocess_input as _preprocess_input from . import get_submodules_from_kwargs from .weights import IMAGENET_WEIGHTS_PATH, IMAGENET_WEIGHTS_HASHES, NS_WEIGHTS_HASHES, NS_WEIGHTS_PATH