Conformer论文以及代码解析(下)

Conformer: Local Features Coupling Global Representations for Visual Recognition

  • 1. main脚本中的参数配置
  • 2. 创建模型
    • 2.1. Conformer类
    • 2.2. ConvBlock类
    • 2.3. Block类
    • 2.4. FCUDown & FCUUp
    • 2.5. ConvTransBlock类
    • 2.6. Conformer的forward整体流程
  • 3. train & val
    • 3.1. loss的计算
    • 3.2. 计算loss以及在val上的acc
  • 4. train的效果图展示
  • 5. Predict效果图展示

上一章节中已经将原论文做了简要的解析,这一章节将解析改论文的源码
代码:

你可能感兴趣的:(pytorch,深度学习,神经网络,计算机视觉,图像识别)