卷积神经网络(Convolutional Neural Networks,简称CNN)是深度学习中最常用的神经网络模型之一。它在计算机视觉领域中取得了巨大的成功,并在图像分类、目标检测、图像生成等任务中发挥着重要作用。本文将介绍一些常见的卷积神经网络模型,并提供相应的源代码。
- LeNet-5
LeNet-5 是一个经典的卷积神经网络模型,由Yann LeCun等人于1998年提出。它主要用于手写数字识别任务。下面是一个简化的 LeNet-5 模型的代码示例:
import torch
import torch.nn as nn
import torch.nn.functional as F
class LeNet5(nn