一文读懂Focal Loss及Pytorch代码(详细注释)

文章目录

  • 前言
  • Focal Loss详解
  • 实现思路
  • Focal Loss类的代码
  • 具体流程分析代码
  • Focal Loss(FL)与CrossEntropy(CE)对比:
  • 总结


前言

Focal Loss及RetinaNet原理见另一篇文章:【论文精读】Focal Loss for Dense Object Detection(RetinaNet)全文翻译及重点总结

本文介绍Focal Loss以及其Pytorch实现。


Focal Loss详解

直接上公式:

你可能感兴趣的:(常见网络的实现,人工智能,深度学习,pytorch,python,神经网络)