计算机视觉系列-轻松掌握 MMDetection 中常用算法 YOLOF(二)

计算机视觉系列-轻松掌握 MMDetection 中常用算法 YOLOF(二)

YOLOF 源码解析

Head

在这里插入图片描述

  • Head 模块是上图的 Decoder 结构,包括分类和回归分支
  • 类似 AutoAssign 算法,在回归分支上并行引入一个 Objectness 分支,用于抑制背景区域的高响应,将其和分类分支相乘。
  • 对外实际上是两个分支,Objectness没有监督 label

mmdetection-2.25.0/mmdet/models/dense_heads/yolof_head.py

  • Objectness 是物体存在于感兴趣区域

猜你喜欢

转载自blog.csdn.net/duan_zhihua/article/details/125679602
今日推荐