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 🚀, AGPL-3.0 license
# YOLOv8-cls image classification model. For Usage examples see https://docs.ultralytics.com/tasks/classify

# Parameters
nc: 11 # number of classes
scales: # model compound scaling constants, i.e. 'model=yolov8n-cls.yaml' will call yolov8-cls.yaml with scale 'n'
  # [depth, width, max_channels]
  n: [0.33, 0.25, 1024]
  s: [0.33, 0.50, 1024]
  m: [0.67, 0.75, 1024]
  l: [1.00, 1.00, 1024]
  x: [1.00, 1.25, 1024]

# YOLOv8.0n backbone
backbone:
  # [from, repeats, module, args]
  - [-1, 1, Mobilenet_v2, [1280]] # 0-P1/2


# YOLOv8.0n head
head:
  - [-1, 1, Classify, [nc]] # Classify

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

相关推荐
AdaCoding2 小时前
YOLOv8架构详解
yolo·网络结构图
云卷云舒___________3 小时前
【Ultralytics YOLO COCO 评估脚本 | 获得COCO评价指标】
yolo·coco·ultralytics
plmm烟酒僧1 天前
在 RK3588 多线程推理 YOLO 时,同时开启硬件解码和 RGA 加速的性能分析
yolo·rkmpp·瑞芯微·硬件加速·rga·色彩空间转换
HABuo2 天前
【YOLOv8】YOLOv8改进系列(11)----替换主干网络之MobileNetV4
人工智能·深度学习·yolo·目标检测·计算机视觉
Kai HVZ2 天前
《深度学习》——yolov4详解
人工智能·深度学习·yolo
plmm烟酒僧2 天前
基于 RK3588 的 YOLO 多线程推理多级硬件加速引擎框架设计(代码框架和实现细节)
yolo·rk3588·多线程·rkmpp·硬件加速·视频解码·librga
Shier833_Ww3 天前
目标识别与双目测距(1)环境搭建:Ubuntu+yolov5+pcl库
linux·yolo·ubuntu
记得开心一点嘛3 天前
YOLOv11模型的常见处理
深度学习·yolo·机器学习
你的陈某某3 天前
基于YOLOv8的PCB缺陷检测--补充实验
yolo·yolov8·pcb检测