小编Adi*_*hil的帖子

我想加载一个我训练过的 keras 模型并将其保存为.pb。
这是代码，

我正在使用 jupyter 笔记本。
模型成功保存为同一目录下的saved_model.pb。但是代码无法访问它。

任何人都可以看到它，我如何访问保存在.pb扩展名中的这个 keras 模型。
我在其他几个地方检查了解决方案，但没有运气。

模型保存在model/saved_model.pb。
我已经取出.pb文件并将其放在我的代码文件所在的同一目录中。

我最近遇到了焦点损失函数，听说它主要用于不平衡数据集。所以我只是使用我在网上找到的这个简单的焦点损失函数（对于 Keras）在 Cifar10 数据集上进行了尝试。

我一直面临着我在最后提到的错误。我尝试了多种方法来解决它，但没有成功。请注意，我非常感谢您的帮助。谢谢你！

焦点损失

import keras.backend as K

ALPHA = 0.8
GAMMA = 2

def FocalLoss(targets, inputs, alpha=ALPHA, gamma=GAMMA):    
    
    inputs = K.flatten(inputs)
    targets = K.flatten(targets)
    
    BCE = K.binary_crossentropy(targets, inputs)
    BCE_EXP = K.exp(-BCE)
    focal_loss = K.mean(alpha * K.pow((1-BCE_EXP), gamma) * BCE)
    
    return focal_loss

输入数据

from keras.datasets import cifar10

(xtrain,ytrain),(xtest,ytest) = cifar10.load_data()

神经网络

from keras.layers import Dense, Conv2D, Flatten, MaxPool2D
from keras.models import Sequential
from keras.optimizers import Adam

model = Sequential([
      Conv2D(filters=64, kernel_size=(27,27), strides=(1,1), input_shape=(32,32,3),padding='same', activation='sigmoid'),
      MaxPool2D(pool_size=(13,13), strides=(1,1), …