如何将CNN模型保存为.h5
时间: 2024-05-16 12:13:46 浏览: 83
基于Tensorflow2.x的CNN的病虫害分类(有界面)
你可以使用Keras模型保存为.h5文件。下面是一个简单的例子:
```python
from keras.models import Sequential
from keras.layers import Dense, Conv2D, Flatten
# 创建模型
model = Sequential()
model.add(Conv2D(32, kernel_size=(3, 3), activation='relu', input_shape=(28,28,1)))
model.add(Flatten())
model.add(Dense(10, activation='softmax'))
# 编译模型
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
# 训练模型
# 保存模型
model.save('my_model.h5')
```
这将保存模型为my_model.h5文件。
阅读全文