一、Detect_DyHead检测头和C2f_DBB模块
详细介绍和代码在往期的博客里:
Detect_DyHead:(YOLOv8改进检测头Detect为Detect_Dyhead-CSDN博客)
C2f_DBB:
(YOLOv8改进之C2f-DBB(C2f模块中融合多元分支模块Diverse Branch Block )-CSDN博客)
二、算法实现
1、将检测头和C2f的模块融合:
ultralytics\ultralytics\nn\other_modules文件夹中要包含DiverseBranchBlock.py和kernel_warehouse.py(开头提到的两篇博客中包含这两个py文件的详细代码)
ultralytics\ultralytics\nn\other_modules\block.py中的代码为:
import torchimport torch.nn as nnimport torch.nn.functional as Ffrom ultralytics.