CenterPoint의 Focal Loss

OpenJR·2023년 10월 23일

Cross Entropy

CE(pt)=(ylog2(pt)+(1y)log2(1pt))CE(p_t)=-(y\cdot\log_2(p_t)+(1-y)\cdot\log_2(1-p_t))

Focal Loss (RetinaNet)

  • γ\gamma: 2
  • α\alpha: 0.25
FL(pt)=(yα(1pt)γlog2(pt)+(1y)(1α)ptγlog2(1pt))FL(p_t)=-(y\cdot\alpha\cdot(1-p_t)^\gamma\cdot\log_2(p_t)+(1-y)\cdot(1-\alpha)\cdot p_t^{\gamma}\cdot\log_2(1-p_t))

Focal Loss (CentetNet, CenterPoint)

FL(pt)=(y(1pt)2log2(pt)+(1y)4ptγlog2(1pt))FL(p_t)=-(y\cdot(1-p_t)^2\cdot\log_2(p_t)+(1-y)^4\cdot p_t^{\gamma}\cdot\log_2(1-p_t))
profile
Jacob

0개의 댓글