计算机视觉系列-YOLOX 在 MMDetection 中复现全流程解析(三)

计算机视觉系列-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)

猜你喜欢

转载自blog.csdn.net/duan_zhihua/article/details/125957572