├── .gitignore ├── LICENSE ├── README.md ├── affinity_loss.py ├── amsoftmax.py ├── conv_ops.py ├── csrc ├── common.hpp ├── cumsum.hpp ├── focal_kernel.cu ├── focal_kernel_old.cu ├── large_margin_kernel.cu ├── layer_norm.cu ├── lovasz_softmax.cu ├── lovasz_softmax_old.cu ├── lsr_kernel.cu ├── mish_kernel.cu ├── ohem_label_kernel.cu ├── one_hot_kernel.cu ├── soft_dice_kernel.cu ├── soft_dice_kernel_v2.cu ├── swish_kernel.cu └── taylor_softmax.cu ├── dice_loss.py ├── dual_focal_loss.py ├── ema.py ├── focal_loss.py ├── generalized_iou_loss.py ├── hswish.py ├── info_nce_dist.py ├── iou_loss.py ├── label_smooth.py ├── large_margin_softmax.py ├── lovasz_softmax.py ├── mish.py ├── one_hot.py ├── partial_fc_amsoftmax.py ├── pc_softmax.py ├── pytorch_loss ├── __init__.py ├── affinity_loss.py ├── amsoftmax.py ├── conv_ops.py ├── dice_loss.py ├── dual_focal_loss.py ├── ema.py ├── focal_loss.py ├── focal_loss_old.py ├── frelu.py ├── group_loss.py ├── hswish.py ├── info_nce_dist.py ├── iou_loss.py ├── label_smooth.py ├── large_margin_softmax.py ├── layer_norm.py ├── lovasz_softmax.py ├── mish.py ├── ohem_loss.py ├── one_hot.py ├── partial_fc_amsoftmax.py ├── pc_softmax.py ├── soft_dice_loss.py ├── swish.py ├── taylor_softmax.py ├── test.py └── triplet_loss.py ├── setup.py ├── soft_dice_loss.py ├── swish.py ├── taylor_softmax.py └── triplet_loss.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/README.md -------------------------------------------------------------------------------- /affinity_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/affinity_loss.py -------------------------------------------------------------------------------- /amsoftmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/amsoftmax.py -------------------------------------------------------------------------------- /conv_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/conv_ops.py -------------------------------------------------------------------------------- /csrc/common.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/common.hpp -------------------------------------------------------------------------------- /csrc/cumsum.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/cumsum.hpp -------------------------------------------------------------------------------- /csrc/focal_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/focal_kernel.cu -------------------------------------------------------------------------------- /csrc/focal_kernel_old.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/focal_kernel_old.cu -------------------------------------------------------------------------------- /csrc/large_margin_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/large_margin_kernel.cu -------------------------------------------------------------------------------- /csrc/layer_norm.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/layer_norm.cu -------------------------------------------------------------------------------- /csrc/lovasz_softmax.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/lovasz_softmax.cu -------------------------------------------------------------------------------- /csrc/lovasz_softmax_old.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/lovasz_softmax_old.cu -------------------------------------------------------------------------------- /csrc/lsr_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/lsr_kernel.cu -------------------------------------------------------------------------------- /csrc/mish_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/mish_kernel.cu -------------------------------------------------------------------------------- /csrc/ohem_label_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/ohem_label_kernel.cu -------------------------------------------------------------------------------- /csrc/one_hot_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/one_hot_kernel.cu -------------------------------------------------------------------------------- /csrc/soft_dice_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/soft_dice_kernel.cu -------------------------------------------------------------------------------- /csrc/soft_dice_kernel_v2.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/soft_dice_kernel_v2.cu -------------------------------------------------------------------------------- /csrc/swish_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/swish_kernel.cu -------------------------------------------------------------------------------- /csrc/taylor_softmax.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/csrc/taylor_softmax.cu -------------------------------------------------------------------------------- /dice_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/dice_loss.py -------------------------------------------------------------------------------- /dual_focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/dual_focal_loss.py -------------------------------------------------------------------------------- /ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/ema.py -------------------------------------------------------------------------------- /focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/focal_loss.py -------------------------------------------------------------------------------- /generalized_iou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/generalized_iou_loss.py -------------------------------------------------------------------------------- /hswish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/hswish.py -------------------------------------------------------------------------------- /info_nce_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/info_nce_dist.py -------------------------------------------------------------------------------- /iou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/iou_loss.py -------------------------------------------------------------------------------- /label_smooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/label_smooth.py -------------------------------------------------------------------------------- /large_margin_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/large_margin_softmax.py -------------------------------------------------------------------------------- /lovasz_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/lovasz_softmax.py -------------------------------------------------------------------------------- /mish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/mish.py -------------------------------------------------------------------------------- /one_hot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/one_hot.py -------------------------------------------------------------------------------- /partial_fc_amsoftmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/partial_fc_amsoftmax.py -------------------------------------------------------------------------------- /pc_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pc_softmax.py -------------------------------------------------------------------------------- /pytorch_loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/__init__.py -------------------------------------------------------------------------------- /pytorch_loss/affinity_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/affinity_loss.py -------------------------------------------------------------------------------- /pytorch_loss/amsoftmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/amsoftmax.py -------------------------------------------------------------------------------- /pytorch_loss/conv_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/conv_ops.py -------------------------------------------------------------------------------- /pytorch_loss/dice_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/dice_loss.py -------------------------------------------------------------------------------- /pytorch_loss/dual_focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/dual_focal_loss.py -------------------------------------------------------------------------------- /pytorch_loss/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/ema.py -------------------------------------------------------------------------------- /pytorch_loss/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/focal_loss.py -------------------------------------------------------------------------------- /pytorch_loss/focal_loss_old.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/focal_loss_old.py -------------------------------------------------------------------------------- /pytorch_loss/frelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/frelu.py -------------------------------------------------------------------------------- /pytorch_loss/group_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/group_loss.py -------------------------------------------------------------------------------- /pytorch_loss/hswish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/hswish.py -------------------------------------------------------------------------------- /pytorch_loss/info_nce_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/info_nce_dist.py -------------------------------------------------------------------------------- /pytorch_loss/iou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/iou_loss.py -------------------------------------------------------------------------------- /pytorch_loss/label_smooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/label_smooth.py -------------------------------------------------------------------------------- /pytorch_loss/large_margin_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/large_margin_softmax.py -------------------------------------------------------------------------------- /pytorch_loss/layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/layer_norm.py -------------------------------------------------------------------------------- /pytorch_loss/lovasz_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/lovasz_softmax.py -------------------------------------------------------------------------------- /pytorch_loss/mish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/mish.py -------------------------------------------------------------------------------- /pytorch_loss/ohem_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/ohem_loss.py -------------------------------------------------------------------------------- /pytorch_loss/one_hot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/one_hot.py -------------------------------------------------------------------------------- /pytorch_loss/partial_fc_amsoftmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/partial_fc_amsoftmax.py -------------------------------------------------------------------------------- /pytorch_loss/pc_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/pc_softmax.py -------------------------------------------------------------------------------- /pytorch_loss/soft_dice_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/soft_dice_loss.py -------------------------------------------------------------------------------- /pytorch_loss/swish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/swish.py -------------------------------------------------------------------------------- /pytorch_loss/taylor_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/taylor_softmax.py -------------------------------------------------------------------------------- /pytorch_loss/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/test.py -------------------------------------------------------------------------------- /pytorch_loss/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/pytorch_loss/triplet_loss.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/setup.py -------------------------------------------------------------------------------- /soft_dice_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/soft_dice_loss.py -------------------------------------------------------------------------------- /swish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/swish.py -------------------------------------------------------------------------------- /taylor_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/taylor_softmax.py -------------------------------------------------------------------------------- /triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CoinCheung/pytorch-loss/HEAD/triplet_loss.py --------------------------------------------------------------------------------