From 7a94c89058561555e021c6c6faf47c6137efdabb Mon Sep 17 00:00:00 2001 From: gtgalone Date: Sat, 9 Jun 2018 17:36:42 +0900 Subject: [PATCH] Fix Keras engine topology to saving --- mrcnn/model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mrcnn/model.py b/mrcnn/model.py index aeb83f3..a97a722 100644 --- a/mrcnn/model.py +++ b/mrcnn/model.py @@ -2094,7 +2094,7 @@ class MaskRCNN(): exlude: list of layer names to excluce """ import h5py - from keras.engine import topology + from keras.engine import saving if exclude: by_name = True @@ -2116,9 +2116,9 @@ class MaskRCNN(): layers = filter(lambda l: l.name not in exclude, layers) if by_name: - topology.load_weights_from_hdf5_group_by_name(f, layers) + saving.load_weights_from_hdf5_group_by_name(f, layers) else: - topology.load_weights_from_hdf5_group(f, layers) + saving.load_weights_from_hdf5_group(f, layers) if hasattr(f, 'close'): f.close() -- GitLab