本文将提供一些关于卷积神经网络(CNN)的代码示例。CNN 是深度学习中用于图像识别、图像分类等任务的重要模型。

代码示例

以下是一个简单的 CNN 模型代码示例:

import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense

# 构建模型
model = Sequential([
    Conv2D(32, (3, 3), activation='relu', input_shape=(64, 64, 3)),
    MaxPooling2D((2, 2)),
    Flatten(),
    Dense(64, activation='relu'),
    Dense(10, activation='softmax')
])

# 编译模型
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])

# 训练模型
model.fit(x_train, y_train, epochs=10)

扩展阅读

想要了解更多关于 CNN 的知识,可以访问以下链接:

TensorFlow Logo