一些适用于分类、分割等网络的损失函数PyTorch实现,包括:

  • label-smooth
  • amsoftmax
  • partial-fc
  • focal-loss
  • dual-focal-loss
  • triplet-loss
  • giou-loss
  • affinity-loss
  • pc_softmax_cross_entropy
  • ohem-loss(softmax based on line hard mining loss)