YOLOv8融合改进 更换检测头为Detect_DyHead同时添加C2f_DBB模块

2023-12-26 18:34:25

一、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 torch
import torch.nn as nn
import torch.nn.functional as F
from ultralytics.

文章来源:https://blog.csdn.net/zmyzcm/article/details/135186381
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。