网站首页
编程语言
数据库
系统相关
其他分享
编程问答
8400
2024-07-06
YOLOV8的Detect head 逐行解读
YOLOV8从不同的特征层,得到不同大小的特征图,然后预测每个特征图的每个格子anchor的类别概率,以及每个格子中物体的边框,即相对于中心点上下左右的偏移量box。shape为[(1,144,80,80),(1,144,40,40),(1,144,20,20)]。 输入x为从不同的上采样层得到的结果x_cat=torch.cat