YOLOV8分类任务更换主干mobilenetv2网络

目录

1.定义模块

 2.导入模块

3.task.py文件更改

4.更改后找文件运行一下

1.定义模块

自定义Movilenet_v2类

class Mobilenet_v2(nn.Module):
    def __init__(self):
        super().__init__()
        model = models.mobilenet_v2(pretrained=True)
        self.layer=nn.Sequential(
        model.features,
        )
    def forward(self, x):
        x=self.layer(x)
        x = nn.functional.adaptive_avg_pool2d(x, (1, 1))
        return x

 2.导入模块

3.task.py文件更改

添加到解析模块

elif m is Mobilenet_v2:
    c2=args[0]
    args=[]

4.更改yaml文件 

 

# Ultralytics YOLO 
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/weixin_40725706/article/detail/992205