yolov8改进之FasterBlock

本文详细介绍了如何使用FasterNet中的FasterBlock替换YOLOv8的C2f中Bottleneck模块,并且使用修改后的yolov8进行目标检测训练与推理。
同样操作,也可替换yolov5中的C2f中的Bottleneck模块。

github地址:https://github.com/JierunChen/FasterNet
paper地址:https://arxiv.org/abs/2303.03667

1、定义C2f_Faster类
在ultralytics/nn/modules/block.py中添加如下代码块,并定义C2f_Faster类:

from timm.models.layers import DropPath

class Partial_conv3(nn.Module):
    def __init__(self

你可能感兴趣的:(模型部署,YOLOv8,FasterBlock,C2f)