yatengLG/Focal-Loss-Pytorch

全中文注释.(The loss function of retinanet based on pytorch).(You can use it on one-stage detection task or classifical task, to solve data imbalance influence).用于one-stage目标检测算法,提升检测效果.你也可以在分类任务中使用该损失函数,解决数据不平衡问题.

语言：Jupyter Notebook · GitHub 星标：496 · Fork：112

在 GitHub 上查看原仓库