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.更改后找文件运行一下

相关推荐
一勺汤4 小时前
YOLOv11改进-双Backbone架构:利用双backbone提高yolo11目标检测的精度
人工智能·yolo·双backbone·double backbone·yolo11 backbone·yolo 双backbone
豆芽8198 小时前
图解YOLO(You Only Look Once)目标检测(v1-v5)
人工智能·深度学习·学习·yolo·目标检测·计算机视觉
Eric.Lee20218 小时前
数据集-目标检测系列- F35 战斗机 检测数据集 F35 plane >> DataBall
人工智能·算法·yolo·目标检测·计算机视觉
白熊1881 天前
【计算机视觉】CV实战项目 - 基于YOLOv5与DeepSORT的智能交通监控系统:原理、实战与优化
人工智能·yolo·计算机视觉
FPGA开源工坊1 天前
FPGA上实现YOLOv5的一般过程
yolo·fpga开发
埃菲尔铁塔_CV算法2 天前
YOLO 模型的深度剖析及其在生物医药领域的创新应用
深度学习·神经网络·yolo·目标检测·计算机视觉
limengshi1383922 天前
使用Python将YOLO的XML标注文件转换为TXT文件格式
xml·python·yolo
unix2linux2 天前
YOLO v5 Series - FFmpeg & (HTML5 + FLV.js ) & ONNX YOLOv5s Integrating
yolo·ffmpeg·html5
宽容人厚载物3 天前
Jetson Orin NX 部署YOLOv12笔记
嵌入式硬件·yolo·jetson·yolov12·英伟达开发板
安步当歌3 天前
【目标检测】对YOLO系列发展的简单理解
图像处理·yolo·目标检测·计算机视觉