yolov8源码解读(part2: 检测,分割)

本文解读下图框中的detection部分 和 没有画出来的分割部分。
yolov8源码解读(part2: 检测,分割)_第1张图片
注意每个模块右上角的数字,它代表第几个模块,
现在检测和分割的输入都是15, 18, 21个模块的输出(图中圆圈处)。
在代码解读的时候会用[15,18,21]提示需要用到15, 18, 21个模块的输出。

yolov8的实例分割用了yolact中的思想,用网络预测出proto和mask coefficient,
它们俩相乘得到分割图,再用检测框crop出实例分割。
不熟悉yolact的可以参考这里。
检测框用了FCOS中的(l, t, r, b)的思路,anchor-free, 不过还是要预先定义目标框中心点。

分割没有画出来,但是它还是有序号的(模块右上角的数字),标为[22].

Segment: [22] nn/modules/head.py

class Segment(Detect):
    

你可能感兴趣的:(源码解读2,YOLO)