我已经构造、拟合并保存了以下模型:
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers
from tensorflow.keras import preprocessing
from tensorflow.keras.models import Sequential
import config
from tensorflow.keras import applications
model = Sequential()
model.add(layers.Flatten(input_shape=input_shape.shape[1:]))
model.add(layers.Dense(100, activation=keras.layers.LeakyReLU(alpha=0.3)))
model.add(layers.Dropout(0.5))
model.add(layers.Dense(50, activation=keras.layers.LeakyReLU(alpha=0.3)))
model.add(layers.Dropout(0.3))
model.add(layers.Dense(num_classes, activation='softmax'))
我正在使用load_model函数进行计算,到目前为止我还没有遇到任何问题,但是我现在得到了以下错误:
ValueError: Unknown activation function: LeakyReLU
我应该在架构上做什么语法上的改变,还是有更深层次的问题?任何建议都将不胜感激,因为我已经尝试设置了一些自定义对象,如下所述:https://github.com/BBQuercus/deepBlink/issues/107
编辑:我在调用load_model的文件中导入的内容如下:
import config
import numpy as np
from tensorflow.keras.preprocessing.image import img_to_array, load_img
from models.create_image_model import make_vgg
import argparse
from tensorflow.keras.models import load_model
import time
from tensorflow import keras
from tensorflow.keras import layers
转载请注明出处:http://www.0591kyj.com/article/20230526/1318291.html