diff --git a/imagenet_utils.py b/imagenet_utils.py index 5cf3d17..2b0c16e 100644 --- a/imagenet_utils.py +++ b/imagenet_utils.py @@ -10,7 +10,7 @@ def preprocess_input(x, dim_ordering='default'): if dim_ordering == 'default': - dim_ordering = K.image_dim_ordering() + dim_ordering = K.common.image_dim_ordering() assert dim_ordering in {'tf', 'th'} if dim_ordering == 'th':