1.检测头原理检测
1.1检测头
在图中的Detect就是yolov8的检测头部分.我们可以发现有两个分支
第一个是
第二个是
分别通过2个3x3的卷积和一个1x1的卷积来提取信息,最后分别结算Bbox.loss和CLs.loss
对应代码部分是cv2和cv3
可以发现,3层卷积之后有一个for循环,来遍历所有的通道数(3通道),这也是大大加大计算量的原因
1.2检测头的计算量
yolov8中的检测头是yolov8中检测目标的关键,它几乎占了计算量的1/5
在图中的Detect就是yolov8的检测头部分.我们可以发现有两个分支
第一个是
第二个是
分别通过2个3x3的卷积和一个1x1的卷积来提取信息,最后分别结算Bbox.loss和CLs.loss
对应代码部分是cv2和cv3
可以发现,3层卷积之后有一个for循环,来遍历所有的通道数(3通道),这也是大大加大计算量的原因
yolov8中的检测头是yolov8中检测目标的关键,它几乎占了计算量的1/5