I'm an algorithm engineer who focused on AI. My main research direction is computer vision and point cloud segmentation. I hope to bring convenience to people through my technology.
yatenglg / focal-loss-pytorch Goto Github PK
View Code? Open in Web Editor NEW全中文注释.(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目标检测算法,提升检测效果.你也可以在分类任务中使用该损失函数,解决数据不平衡问题.
Home Page: https://ptorch.com/news/253.html