YOLO 后处理之坐标变换

YOLO 后处理之坐标变换#

YOLO 后处理的坐标变换主要目的是将模型预测的坐标转换为图像上的真实坐标。

  1. 模型预测的坐标表示 YOLO 模型通常会对每个预测框输出 4 个坐标值,一般为 (tx,ty,tw,th)(t_x, t_y, t_w, t_h),这些值是相对于网格单元的偏移和缩放。

    • txt_xtyt_y:预测框中心相对于所在网格单元左上角的偏移。

    • twt_wtht_h:预测框的宽度和高度相对于锚框的缩放。

  2. 从网格单元偏移到特征图坐标 将预测的偏移和缩放转换为特征图上的坐标 (bx,by,bw,bh)(b_x, b_y, b_w, b_h)

    bx=σ(tx)+cxby=σ(ty)+cybw=pwexp(tw)bh=phexp(th) \begin{aligned} &b_x = \sigma(t_x) + c_x\\ &b_y = \sigma(t_y) + c_y\\ &b_w = p_w * \exp(t_w)\\ &b_h = p_h * \exp(t_h) \end{aligned}

    其中:

    • σ\sigma 是 Sigmoid 函数, 用于将偏移限制在 [0,1][0, 1] 范围内。

    • cxc_xcyc_y 是当前网格单元左上角在特征图上的坐标。

    • pwp_wphp_h 是预先定义的锚框的宽度和高度。

  3. 从特征图坐标到图像坐标 将特征图上的坐标转换为原始图像上的坐标 (x,y,w,h)(x, y, w, h)

    x=bxstridey=bystridew=bwstrideh=bhstride \begin{aligned} &x = b_x * \text{stride}\\ &y = b_y * \text{stride}\\ &w = b_w * \text{stride}\\ &h = b_h * \text{stride}\\ \end{aligned}

    其中,stride 是特征图相对于原始图像的下采样倍数。