2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added custom objects to recover saved model

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-09 12:13:41 +02:00
parent f5c8e9483a
commit 4bdd069650
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/twomartens/masterthesis/cli.py
View File

@ -307,6 +307,9 @@ def _ssd_test(args: argparse.Namespace) -> None:
from twomartens.masterthesis import data
from twomartens.masterthesis import ssd
from twomartens.masterthesis.ssd_keras.keras_layers import keras_layer_AnchorBoxes
from twomartens.masterthesis.ssd_keras.keras_layers import keras_layer_L2Normalization
config = tf.ConfigProto()
config.log_device_placement = False
@ -334,7 +337,10 @@ def _ssd_test(args: argparse.Namespace) -> None:
instances = pickle.load(file)
# model
ssd_model = tf.keras.models.load_model(model_file)
ssd_model = tf.keras.models.load_model(model_file, custom_objects={
"L2Normalization": keras_layer_L2Normalization.L2Normalization,
"AnchorBoxes": keras_layer_AnchorBoxes.AnchorBoxes
})
test_generator, length_dataset = \
data.load_scenenet_data(file_names_photos, instances, coco_path,