本文详细介绍了如何使用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