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