卷积神经网络 (Convolutional Neural Network, CNN) 是一种常用的深度学习模型,用于图像分类、物体检测和其他计算机视觉任务。
下面是一个简单的 CNN 的 Python 代码示例:
```python import tensorflow as tf
定义卷积层
def conv_layer(inputs, filters, kernel_size, strides, padding, activation=tf.nn.relu): layer = tf.keras.layers.Conv2D(filters=filters, kernel_size=kernel_size, strides=strides, padding=padding)(inputs) if activation: layer = tf.keras.layers.Activation(activation)(layer) return layer
定义池化层
def pool_layer(inputs, pool_size, strides, padding): return tf.keras.layers.MaxPool2D(pool_size=pool_size, strides=strides, padding=padding)(inputs)
定义全连接层
def dense_layer(inputs, units, activation=tf.nn.relu): layer = tf.keras.layers.Dense(units=units)(inputs)