文件名称：分类，分割等网络所需的相关loss pytorch
文件大小：116KB
文件格式：ZIP
更新时间：2022-02-17 08:20:29
分类 网络 pytorch 数据挖掘 人工智能
My implementation of 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)
【文件预览】：
pytorch-loss-master
----amsoftmax.py(2KB)
----taylor_softmax.py(8KB)
----dual_focal_loss.py(1KB)
----triplet_loss.py(1024B)
----info_nce_dist.py(5KB)
----hswish.py(4KB)
----affinity_loss.py(4KB)
----conv_ops.py(3KB)
----one_hot.py(3KB)
----pytorch_loss()
--------amsoftmax.py(1KB)
--------frelu.py(557B)
--------taylor_softmax.py(9KB)
--------dual_focal_loss.py(1KB)
--------triplet_loss.py(1024B)
--------info_nce_dist.py(5KB)
--------hswish.py(4KB)
--------layer_norm.py(9KB)
--------test.py(2KB)
--------affinity_loss.py(4KB)
--------conv_ops.py(3KB)
--------one_hot.py(3KB)
--------soft_dice_loss.py(7KB)
--------label_smooth.py(8KB)
--------lovasz_softmax.py(8KB)
--------mish.py(4KB)
--------ohem_loss.py(2KB)
--------focal_loss_old.py(8KB)
--------__init__.py(1KB)
--------partial_fc_amsoftmax.py(8KB)
--------focal_loss.py(8KB)
--------dice_loss.py(6KB)
--------large_margin_softmax.py(10KB)
--------swish.py(4KB)
--------generalized_iou_loss.py(1KB)
--------ema.py(3KB)
--------pc_softmax.py(6KB)
----soft_dice_loss.py(7KB)
----label_smooth.py(8KB)
----lovasz_softmax.py(8KB)
----mish.py(4KB)
----LICENSE(1KB)
----partial_fc_amsoftmax.py(8KB)
----focal_loss.py(8KB)
----setup.py(1KB)
----.gitignore(1KB)
----dice_loss.py(6KB)
----large_margin_softmax.py(10KB)
----swish.py(4KB)
----README.md(3KB)
----csrc()
--------cumsum.hpp(5KB)
--------swish_kernel.cu(8KB)
--------focal_kernel.cu(7KB)
--------one_hot_kernel.cu(11KB)
--------lsr_kernel.cu(11KB)
--------lovasz_softmax_old.cu(24KB)
--------mish_kernel.cu(4KB)
--------large_margin_kernel.cu(20KB)
--------layer_norm.cu(13KB)
--------soft_dice_kernel_v2.cu(16KB)
--------focal_kernel_old.cu(7KB)
--------soft_dice_kernel.cu(10KB)
--------common.hpp(2KB)
--------taylor_softmax.cu(15KB)
--------ohem_label_kernel.cu(9KB)
--------lovasz_softmax.cu(28KB)
----generalized_iou_loss.py(1KB)
----ema.py(3KB)
----pc_softmax.py(6KB)