电脑基础 · 2023年4月19日

优化改进YOLOv5算法之添加SE、CBAM、CA模块(超详细)

目录

1 SENet

1.1 SENet原理

1.2SENet代码(Pytorch)

1.3YOLOv5中加入SE模块

1.3.1common.py配置

1.3.2yolo.py配置