jog*_*gan 3 python neural-network deep-learning keras
我有一个简单的keras模型。保存模型后。我无法加载模型。这是我实例化模型并尝试加载权重后得到的错误:
Using TensorFlow backend.
Traceback (most recent call last):
File "test.py", line 4, in <module>
model = load_model("test.h5")
File "/usr/lib/python3.7/site-packages/keras/engine/saving.py", line 419, in load_model
model = _deserialize_model(f, custom_objects, compile)
File "/usr/lib/python3.7/site-packages/keras/engine/saving.py", line 258, in _deserialize_model
.format(len(layer_names), len(filtered_layers))
ValueError: You are trying to load a weight file containing 6 layers into a model with 0 layers
Run Code Online (Sandbox Code Playgroud)
用于实例化模型并使用model.load_weights进行模型汇总。当我使用print(model)打印模型时,我什么也没有
Traceback (most recent call last):
File "test.py", line 7, in <module>
print(model.summary())
AttributeError: 'NoneType' object has no attribute 'summary'
Run Code Online (Sandbox Code Playgroud)
这是我的网络:
from keras.models import Sequential
from keras.layers import Conv2D, MaxPooling2D, InputLayer, Flatten, Dense, BatchNormalization
def create_model():
kernel_size = 5
pool_size = 2
batchsize = 64
model = Sequential()
model.add(InputLayer((36, 120, 1)))
model.add(Conv2D(filters=20, kernel_size=kernel_size, activation='relu', padding='same'))
model.add(BatchNormalization())
model.add(MaxPooling2D(pool_size))
model.add(Conv2D(filters=50, kernel_size=kernel_size, activation='relu', padding='same'))
model.add(BatchNormalization())
model.add(MaxPooling2D(pool_size))
model.add(Flatten())
model.add(Dense(120, activation='relu'))
model.add(Dense(2, activation='relu'))
return model
Run Code Online (Sandbox Code Playgroud)
培训程序脚本:
import numpy as np
from keras import optimizers
from keras import losses
from sklearn.model_selection import train_test_split
from model import create_model
def data_loader(images, pos):
while(True):
for i in range(0, images.shape[0], 64):
if (i+64) < images.shape[0]:
img_batch = images[i:i+64]
pos_batch = pos[i:i+64]
yield img_batch, pos_batch
else:
img_batch = images[i:]
pos_batch = pos[i:]
yield img_batch, pos_batch
def main():
model = create_model()
sgd = optimizers.Adadelta(lr=0.01, rho=0.95, epsilon=None, decay=0.0)
model.compile(loss=losses.mean_squared_error, optimizer=sgd)
print("traning")
data = np.load("data.npz")
images = data['images']
pos = data['pos']
x_train, x_test, y_train, y_test = train_test_split(images, pos, test_size=0.33, random_state=42)
model.fit_generator(data_loader(x_train, y_train), steps_per_epoch=x_train.shape[0]//64, validation_data=data_loader(x_test, y_test), \
validation_steps = x_test.shape[0]//64, epochs=1)
model.save('test.h5')
model.save_weights('test_weights.h5')
print("training done")
if __name__ == '__main__':
main()
Run Code Online (Sandbox Code Playgroud)
在第一层中拖放InputLayer使用input_shape。您的代码将类似于:
model = Sequentional()
model.add(Conv2D(filters=20,..., input_shape=(36, 120, 1)))Run Code Online (Sandbox Code Playgroud)
似乎与InputLayer的模型未HDF5正确序列化。
将您的Tensorflow和Keras升级到最新版本
修复解释问题,解释这里