各位哥哥姐姐弟弟妹妹大家好,我是干饭王刘姐,主业干饭,主业2.0计算机研究生在读。
和我一起来改进yolov8变身计算机大牛吧!
本文中的论文笔记都是刘姐亲自整理,原创整理哦~
小目标检测层简介
在标准目标检测中,数据集中有小目标的情况下,小目标会出现漏检或检测效果不佳等问题。YOLO默认有3个检测头,能够多尺度对目标行检测其检测尺寸分别是:
P3/8 对应的检测特征图大小为80X80,用于检测大小在8X8以上的目标。
P4/16对应的检测特征图大小为40X40,用于检测大小在16X16以上的目标。
P5/32对应的检测特征图大小为20X20,用于检测大小在32X32以上的目标。
但对微小目标检测可能存在检测能力不佳的现象,因此添加一个微小物体的检测头,在有小目标数据集中能够有涨点。
新增加160X160的检测特征图,用于检测4X4以上的目标。
yaml
# Ultralytics YOLO 🚀, AGPL-3.0 license
# YOLOv8 object detection model with P2-P5 outputs. For Usage examples see https://docs.ultralytics.com/tasks/detect
# Parameters
nc