深度学习是人工智能领域的核心,而卷积神经网络(CNN)在图像识别、图像处理等领域有着广泛的应用。本教程将介绍如何使用 TensorFlow 构建 CNN。
基础概念
- 卷积神经网络(CNN):一种特殊的人工神经网络,擅长于图像识别和图像处理。
- TensorFlow:一个开源的深度学习框架,由 Google 工程师开发。
安装 TensorFlow
pip install tensorflow
实例教程
以下是一个简单的 TensorFlow CNN 模型示例,用于识别猫和狗的图片:
import tensorflow as tf
# 构建模型
model = tf.keras.Sequential([
tf.keras.layers.Conv2D(32, (3, 3), activation='relu', input_shape=(64, 64, 3)),
tf.keras.layers.MaxPooling2D((2, 2)),
tf.keras.layers.Conv2D(64, (3, 3), activation='relu'),
tf.keras.layers.MaxPooling2D((2, 2)),
tf.keras.layers.Conv2D(128, (3, 3), activation='relu'),
tf.keras.layers.MaxPooling2D((2, 2)),
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(128, activation='relu'),
tf.keras.layers.Dense(1, activation='sigmoid')
])
# 编译模型
model.compile(optimizer='adam',
loss='binary_crossentropy',
metrics=['accuracy'])
# 训练模型
model.fit(train_images, train_labels, epochs=10)
# 评估模型
test_loss, test_acc = model.evaluate(test_images, test_labels, verbose=2)
扩展阅读
想要深入了解 TensorFlow 和 CNN?可以参考以下资源:
TensorFlow 图标