YOLOv10改进 | 检测头篇 | YOLOv10引入四头检测头

小李学AI

已于 2024-06-25 15:47:04 修改

阅读量1.2k

点赞数 3

分类专栏： YOLOv10有效涨点专栏文章标签： YOLO 目标检测计算机视觉人工智能深度学习机器学习

于 2024-06-24 10:09:42 首次发布

本文链接：https://blog.csdn.net/tsg6698/article/details/139915154

版权

YOLOv10有效涨点专栏专栏收录该内容

102 篇文章 64 订阅 ¥89.90 ¥99.00

订阅专栏

1. 改后结构图：

2. 创建一个yolov10nsitou.yaml

# Parameters
nc: 2 # number of classes
scales: # model compound scaling constants, i.e. 'model=yolov8n.yaml' will call yolov8.yaml with scale 'n'
  # [depth, width, max_channels]
  n: [0.33, 0.25, 1024] 

# YOLOv8.0n backbone
backbone:
  # [from, repeats, module, args]
  - [-1, 1, Conv, [64, 3, 2]] # 0-P1/2
  - [-1, 1, Conv, [128, 3, 2]] # 1-P2/4
  - [-1, 3, C2f, [128, True]]
  - [-1, 1, Conv, [256, 3, 2]] # 3-P3/8
  - [-1, 6, C2f, [256, True]]
  - [-1, 1, SCDown, [512, 3, 2]] # 5-P4/16
  - [-1, 6, C2f, [512, True]]
  - [-1, 1, SCDown, [1024, 3,

了解本专栏