This commit is contained in:
Ankle committed 2026-09-21 07:22:28 +00:00
1 parent c646ca2895
commit 0878dedb19
1 file changed
+15 -15
+15 -15
View File
@@ -1,24 +1,24 @@
这是张伟的本科毕业设计。基于yolov5s6模型,对其output进行了改进,增强yolov5s6的分类能力,使得这个模型能同时较好地完成“目标检测+分类”这两个任务。
M0_yolov5_raw:
原生yolov5s6的output
![YOLOv5 Baseline](docs/images/1、YOLOv5-baseline.png)
M0_yolov5_raw:原生yolov5s6的output
G8_decoupled_head_light分支:
解耦output,将目标检测和分类任务的output拆开,便于后续单独增强分类的能力
![YOLOv5s6 Decoupled Head](docs/images/2、YOLOv5s6-DecoupledHead.png)
![YOLOv5 Baseline](./docs/images/1、YOLOv5-baseline.png)
G11_cls_head_enhance_Lightweight分支:
在分类支添加两层卷积
![YOLOv5s6 Head Variant](docs/images/3、YOLOv5s6-HeadVariant.png)
G8_decoupled_head_light分支:解耦output,将目标检测和分类任务的output拆开,便于后续单独增强分类的能力
G13_decoupled_head_ECA分支:
在分类支添加ECA通道注意力机制
![YOLOv5s6 Decoupled Head ECA](docs/images/4、YOLOv5s6-DecoupledHead-ECA.png)
![YOLOv5s6 Decoupled Head](./docs/images/2、YOLOv5s6-DecoupledHead.png)
G12_cls_head_enhance_and_ECA分支:
在分类支同时添加两层卷积和ECA
![YOLOv5s6 Head Variant ECA](docs/images/5、YOLOv5s6-HeadVariant-ECA.png)
G11_cls_head_enhance_Lightweight分支:在分类支添加两层卷积
![YOLOv5s6 Head Variant](./docs/images/3、YOLOv5s6-HeadVariant.png)
G13_decoupled_head_ECA分支:在分类支添加ECA通道注意力机制
![YOLOv5s6 Decoupled Head ECA](./docs/images/4、YOLOv5s6-DecoupledHead-ECA.png)
G12_cls_head_enhance_and_ECA分支:在分类支同时添加两层卷积和ECA
![YOLOv5s6 Head Variant ECA](./docs/images/5、YOLOv5s6-HeadVariant-ECA.png)
以下是yolov5原git仓的readme