计算机视觉系列-YOLOX 在 MMDetection 中复现全流程解析(三)
YoloX 简介
论文:YOLOX: Exceeding YOLO Series in 2021
论文链接:https://arxiv.org/abs/2107.08430
计算 loss
旷视科技loss源码分析
网络输出包括 3 个 尺度, stride 分别是 8、16 和 32,每个输出尺度上又包括 3 个输出,分别是 bbox 输出分支、objectness 输出分支和 cls 类别输出分支
-
计算 3 个输出层所需要的特征图尺度的坐标,用于 bbox 解码
megvii_YOLOX_main/yolox/models/yolo_head.py
def get_output_and_grid(self, output, k, stride, dtype)