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

Fixed not-compiled model

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-09 12:53:15 +02:00
parent 9b13c5cf52
commit 38ce65d367
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -309,6 +309,7 @@ def _ssd_test(args: argparse.Namespace) -> None:
from twomartens.masterthesis import ssd 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_AnchorBoxes
from twomartens.masterthesis.ssd_keras.keras_layers import keras_layer_L2Normalization from twomartens.masterthesis.ssd_keras.keras_layers import keras_layer_L2Normalization
from twomartens.masterthesis.ssd_keras.keras_loss_function import keras_ssd_loss
config = tf.ConfigProto() config = tf.ConfigProto()
@ -341,6 +342,18 @@ def _ssd_test(args: argparse.Namespace) -> None:
"L2Normalization": keras_layer_L2Normalization.L2Normalization, "L2Normalization": keras_layer_L2Normalization.L2Normalization,
"AnchorBoxes": keras_layer_AnchorBoxes.AnchorBoxes "AnchorBoxes": keras_layer_AnchorBoxes.AnchorBoxes
}) })
# TODO finde clean solution rather than Copy & Paste
learning_rate_var = tf.keras.backend.variable(conf.get_property("Parameters.learning_rate"))
ssd_loss = keras_ssd_loss.SSDLoss()
ssd_model.compile(
optimizer=tf.train.AdamOptimizer(learning_rate=learning_rate_var,
beta1=0.9, beta2=0.999),
loss=ssd_loss.compute_loss,
metrics=[
"categorical_accuracy"
]
)
test_generator, length_dataset = \ test_generator, length_dataset = \
data.load_scenenet_data(file_names_photos, instances, coco_path, data.load_scenenet_data(file_names_photos, instances, coco_path,