kuangliu / pytorch-retinanet

RetinaNet in PyTorch
992 stars 250 forks source link

PyTorch-RetinaNet

Train RetinaNet with Focal Loss in PyTorch.

Reference:
[1] Focal Loss for Dense Object Detection