name 'backend' is not defined

最新推荐文章于 2022-10-09 17:11:40 发布

AI剑客

最新推荐文章于 2022-10-09 17:11:40 发布

阅读量1.7k

点赞数 1

分类专栏： AI keras

本文链接：https://blog.csdn.net/qq_43258953/article/details/103113680

版权

AI 同时被 2 个专栏收录

145 篇文章 12 订阅

订阅专栏

keras

37 篇文章 1 订阅

订阅专栏

在kears模型中有用到backend，在训练时没有异常，在load model进行predict时，居然报错：name ‘backend’ is not defined。

import keras.backend as backend

y = Lambda(lambda z: z[:, :, :, j * c:(j + 1) * c]
     if backend.image_data_format() == 'channels_last' else
            lambda z: z[:, j * c:(j + 1) * c, :, :])(x)

解决方式如下：
在loadmodel时，自定义字典custom_objects里面加入‘backend’:backend

import keras.backend as backend

model = load_model('./unet_' + label + '_20.h5',
                   custom_objects={'focal_loss_fixed': focal_loss(),
‘backend':backend},compile=False)