计算机视觉 之 经典模型汇总

文章目录

早期特征工程模型

SIFT(尺度不变特征变换) 2004


HOG(方向梯度直方图) 2005

Haar特征 + AdaBoost(2001)

深度学习模型

LeNet-5 (1998)

  • 地位:首个成功应用于数字识别的卷积神经网络(CNN),奠定了 CNN 的基础架构。
  • 结构:包含卷积层、池化层、全连接层,输入为 32×32 的手写数字图像。
  • 特点:
    • 首次验证了卷积操作(局部感受野、权值共享)在图像处理中的优势,大幅减少参数数量。
    • 仅适用于简单任务(如 MNIST 手写数字识别),对复杂图像效果有限。

AlexNet (2012)

  • 地位:深度学习引爆计算机视觉的标志性模型,在 ImageNet 竞赛中准确率远超传统方法。
  • 结构:8 层网络(5 层卷积 + 3 层全连接) ,首次使用 ReLU 激活函数、Dropout 数据增强
  • 特点:
    • 突破了传统神经网络的计算瓶颈,证明了深度 CNN 在大规模图像识别中的潜力。
    • 引入 GPU 加速训练,推动了深度学习硬件和软件的发展。




python 复制代码
import torch.nn as nn 
# AlexNet 网络结构
# 其实就是 前面5个卷积层,后面3个全连接层
# 卷积层:
# 1. 卷积层1:输入3通道,输出96个特征图,卷积核大小11x11,步长4,填充2
# 2. 卷积层2:输入96个特征图,输出256个特征图,卷积核大小5x5,步长2,填充2
# 3. 卷积层3:输入256个特征图,输出384个特征图,卷积核大小3x3,步长1,填充1
# 4. 卷积层4:输入384个特征图,输出384个特征图,卷积核大小3x3,步长1,填充1
# 5. 卷积层5:输入384个特征图,输出256个特征图,卷积核大小3x3,步长1,填充1
# 全连接层:
# 1. 全连接层1:输入256x5x5,输出4096
# 2. 全连接层2:输入4096,输出4096
# 3. 全连接层3:输入4096,输出10
class AlexNet(nn.Module):
    def __init__(self):
        super(AlexNet, self).__init__()
        self.conv = nn.Sequential(
            nn.Conv2d(3,96,kernel_size=11,stride=4),
            nn.ReLU(),
            nn.MaxPool2d(kernel_size=3,stride=2),
            # 减小卷积窗口
            nn.Conv2d(96,256,kernel_size=5,padding=2),
            nn.ReLU(),
            nn.MaxPool2d(kernel_size=3,stride=2),
            # 连续三个卷积层
            nn.Conv2d(256,384,kernel_size=3,padding=1),
            nn.ReLU(),
            nn.Conv2d(384,384,kernel_size=3,padding=1),
            nn.ReLU(),
            nn.Conv2d(384,256,kernel_size=3,padding=1),
            nn.ReLU(),
            nn.MaxPool2d(kernel_size=3,stride=2)
        )
        self.fc = nn.Sequential(
            nn.Linear(256*5*5,4096),
            nn.ReLU(),
            nn.Dropout(0.5),
            nn.Linear(4096,4096),
            nn.ReLU(),
            nn.Dropout(0.5),
            nn.Linear(4096,10)
        )
    def forward(self,img):
        feature = self.conv(img)
        output = self.fc(feature.view(img.shape[0],-1))
        return output

分析

  • ReLU激活函数,一般在这个卷积层全连接层后面,用于:

    • 解决梯度消失的问题
    • 使得神经网络可以学习到这个复杂的非线性映射
  • Dropout 这个全连接层使用(卷积层参数共享并且参数量比较少),训练的时候启用,但是推理的时候关闭(PyTorch 会自动处理,通过model.train()和model.eval()控制

    • 通过随机丢弃神经元,减少神经元间的共适应,降低过拟合。
    • 相当于集成多个子网络,提升泛化能力

转置卷积

  • 作用:
    • CNN 可视化:可将卷积得到的特征图还原到像素空间,用于观察特征图对哪些模式响应最大,从而可视化卷积操作提取出的特征。
    • 图像分割:在全卷积网络(FCN)中,需要对图像进行像素级的分割,通过反卷积将特征图尺寸还原到原来的大小,实现上采样操作。
    • 生成模型:在生成对抗网络(GAN)中,需要从输入向量生成图像,通过反卷积将提取的特征图还原到和原图同样尺寸的大小,生成最终的图像。
  • 与卷积的区别:卷积通常会使图像尺寸变小,用于提取特征;而反卷积主要用于将特征图尺寸扩大,实现上采样。虽然反卷积看上去像是正向卷积的逆运算,但它只能还原原始图像的尺寸不能真的恢复原始图像内容,即输出特征图的每个元素值与原始输入图像对应位置的元素值是不一样的。
相关推荐
kikikidult3 小时前
Ubuntu20.04运行openmvg和openmvs实现三维重建(未成功,仅供参考)
人工智能·笔记·ubuntu·计算机视觉
189228048613 小时前
NW728NW733美光固态闪存NW745NW746
大数据·服务器·网络·人工智能·性能优化
大模型最新论文速读4 小时前
模拟注意力:少量参数放大 Attention 表征能力
人工智能·深度学习·机器学习·语言模型·自然语言处理
lishaoan774 小时前
用TensorFlow进行逻辑回归(二)
人工智能·tensorflow·逻辑回归
慌ZHANG5 小时前
智慧气象新范式:人工智能如何重构城市级气象服务生态?
人工智能
Eumenidus5 小时前
使用ESM3蛋白质语言模型进行快速大规模结构预测
人工智能·语言模型·自然语言处理
熊猫钓鱼>_>5 小时前
FastGPT革命:下一代语言模型的极速进化
人工智能·语言模型·自然语言处理
吕永强5 小时前
电网的智能觉醒——人工智能重构能源生态的技术革命与公平悖论
人工智能·科普
极限实验室5 小时前
喜报 - 极限科技荣获 2025 上海开源创新菁英荟「开源创新新星企业」奖
人工智能·开源
在美的苦命程序员5 小时前
芯片之后,AI之争的下一个战场是能源?
人工智能