源码屋整站源码,做百度移动网站吗,科技服务网站建设方案,网页升级紧急通知页面1、前言
本文会将YOLOV5 backbone更换成Swin Transformer 具体为什么这样实现参考上文:YOLOV5 改进:替换backbone(MobileNet为例)-CSDN博客
这里只贴加入的代码
训练结果如下: 2、common文件更改
在common文件中加入下面代码:
这里是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