您现在的位置是：首页 >技术杂谈 >YOLOv5/v7 更换骨干网络之 SwinTransformer网站首页技术杂谈

YOLOv5/v7 更换骨干网络之 SwinTransformer

迪菲赫尔曼 2023-05-21 16:00:02
简介YOLOv5/v7 更换骨干网络之 SwinTransformer

提供 YOLOv5 / YOLOv7 / YOLOv7-tiny 模型 YAML 文件

在这里插入图片描述

论文地址：https://arxiv.org/pdf/2103.14030.pdf
代码地址：https://github.com/microsoft/Swin-Transformer

本文介绍了一种新的视觉Transformer，称为Swin Transformer，它可以作为计算机视觉通用的骨干网络。从语言到视觉的转换中，适应Transformer所面临的挑战源于两个领域之间的差异，如视觉实体尺度的巨大变化和图像中像素的高分辨率与文本中单词的差异。为了解决这些差异，我们提出了一种分层Transformer，其表示是通过Shifted窗口计算的。Shifted窗口方案通过将自注意计算限制在非重叠的本地窗口内，同时允许跨窗口连接，从而提高了效率。这种分层架构具有在不同尺度下进行建模的灵活性，并且与图像大小的计算复杂度呈线性关系。这些特性使Swin Transformer与广泛的视觉任务兼容，包括图像分类（在ImageNet-1K上的87.3的top

风语者！平时喜欢研究各种技术，目前在从事后端开发工作，热爱生活、热爱工作。

站长推荐