1、前言

本文会将YOLOV5 backbone更换成Swin Transformer

具体为什么这样实现参考上文：YOLOV5 改进：替换backbone（MobileNet为例）-CSDN博客

这里只贴加入的代码

训练结果如下：

 

2、common文件更改

在common文件中加入下面代码：

这里是swin transformer的实现，参考：Swin-Transformer 在图像识别中的应用_swim transformer 应用-CSDN博客

""" 
Swin Transformer and YOLOV5
"""

import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.utils.checkpoint as checkpoint
import numpy a