tensorflow - 加载 tensorflow.keras 模型时如何加载损失函数？

Question

我正在尝试加载保存的 tensorflow.keras 模型，加载时出现以下错误。

ValueError: Unknown loss function:cce_dice_loss

cce_dice_loss 来自库 segmentation_models

请找到以下损失函数的代码

from segmentation_models.losses import cce_dice_loss

model2.compile(optimizer, cce_dice_loss, metrics=[iou_score])

请找到以下用于保存和加载模型的代码

model2.save("my_model",save_format='tf')
new_model = tf.keras.models.load_model('my_model', custom_objects={'convolutional_block': convolutional_block,'identity_block':identity_block,'global_flow':global_flow,'context_flow':context_flow,'sum_layer':sum_layer,'fsm':fsm,'agcn':agcn,'iou_score':iou_score,'focal_loss':focal_loss})

在加载模型时，我遇到了我提到的错误，请找到错误的屏幕

谁能帮我解决这个问题。

标签： tensorflowkeras