计算机视觉系列-YOLOX 在 MMDetection 中复现全流程解析（四）

计算机视觉系列-YOLOX 在 MMDetection 中复现全流程解析（四）

YoloX 简介

论文：YOLOX: Exceeding YOLO Series in 2021
论文链接：https://arxiv.org/abs/2107.08430

MMDetection 重构Loss

上一篇博客分析了旷视科技YOLOX Loss的源码分析，本文学习MMDetection YoloX Loss的复现重构，采用 MMDetection 中 的构建方式： prior_generator + bbox assign + bbox encode decode + loss

• YOLOX 是 anchor-free 算法， 仍需要特征图上每个预测点的 point 信息，采用 RepPoints 中的 MlvlPointGenerator 类生成 anchor-point 坐标
• bbox assign 新建了 SimOTAAssigner 类，用于对预测点进行分配正负样本；

@HEADS.register_module()
class