YOLOV5 改进：替换backbone为Swin Transformer
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原文地址：https://blog.csdn.net/qq_44886601/article/details/136919487
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！