From 2561ad94de9cf1ff84a3b89069130dbb2f19a531 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Radoslav=20Pito=C5=88=C3=A1k?= Date: Thu, 27 Jan 2022 17:19:07 +0100 Subject: [PATCH] [Design_Tutorials/05-Keras_FCN8_UNET_segmentation] epochs in model name --- .../files/code/Keras2TF.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/Design_Tutorials/05-Keras_FCN8_UNET_segmentation/files/code/Keras2TF.py b/Design_Tutorials/05-Keras_FCN8_UNET_segmentation/files/code/Keras2TF.py index 2fde18a1..afd978ba 100644 --- a/Design_Tutorials/05-Keras_FCN8_UNET_segmentation/files/code/Keras2TF.py +++ b/Design_Tutorials/05-Keras_FCN8_UNET_segmentation/files/code/Keras2TF.py @@ -66,15 +66,15 @@ # load weights & architecture into new model if model_name=="fcn8ups" : - weights= "fcn8ups/ep200_trained_fcn8ups_224x224.hdf5" + weights= f"fcn8ups/ep{cfg.EPOCHS}_trained_fcn8ups_224x224.hdf5" elif model_name=="fcn8" : - weights= "fcn8/ep200_trained_fcn8_224x224.hdf5" + weights= f"fcn8/ep{cfg.EPOCHS}_trained_fcn8_224x224.hdf5" elif model_name=="unet1" : - weights= "unet/ep200_trained_unet_model1_224x224.hdf5" + weights= f"unet/ep{cfg.EPOCHS}_trained_unet_model1_224x224.hdf5" elif model_name=="unet2" : - weights= "unet/ep200_trained_unet_model2_224x224.hdf5" + weights= f"unet/ep{cfg.EPOCHS}_trained_unet_model2_224x224.hdf5" else: # elif model_name=="unet3" : - weights= "unet/ep200_trained_unet_model3_224x224.hdf5" + weights= f"unet/ep{cfg.EPOCHS}_trained_unet_model3_224x224.hdf5" print("model name = ", model_name) filename = os.path.join(WEIGHTS_DIR,weights)