YOLOv5s-set中添加Swin-Transformer后的架构框图
时间: 2023-06-13 08:02:52 浏览: 194
以下是YOLOv5s-set中添加Swin-Transformer后的架构框图示意图:
```
+------------------------------------------------------------------------+
| |
| Backbone |
| |
| +------------------------+ +------------------------+ |
| | Stem | | Swin-Transformer | |
| +-----------+------------+ +-----------+------------+ |
| | | |
| +-----------v------------+ +-----------v------------+ |
| | Body | | Swin-Transformer | |
| +-----------+------------+ +-----------+------------+ |
| | | |
| +-----------v------------+ +-----------v------------+ |
| | Head | | Swin-Transformer | |
| +------------------------+ +------------------------+ |
| |
+------------------------------------------------------------------------+
```
其中,Backbone包含了Stem、Body和Head三个部分。在原有的YOLOv5s-set中,Backbone主要由CSPDarknet53实现,而在添加Swin-Transformer后,Swin-Transformer取代了CSPDarknet53的位置,同时也被拆分为了多个部分,分别嵌入在了Stem、Body和Head中。总体上,这个框架还是沿用了YOLOv5s-set的设计思路,只不过将其中的CSPDarknet53替换为了Swin-Transformer。
阅读全文
相关推荐
















