【文件属性】：
文件名称：pytorch-loss:标签平滑，amsoftmax，散焦，三重损失，lovasz-softmax。 也许有用
文件大小：93KB
文件格式：ZIP
更新时间：2021-04-30 16:41:15
cuda pytorch ema triplet-loss one-hot
火炬损失
我实现的标签平滑，amsoftmax，焦点损耗，双焦点损耗，三重态损耗，giou损耗，亲和力损耗，pc_softmax_cross_entropy，ohem损耗（基于行硬挖掘损失的softmax），大利润- softmax（bmvc2019），lovasz-softmax-loss和dice-loss（广义的软骰子损失和批处理软骰子损失）。 也许这对我的未来工作很有用。
还尝试实现swish，hard-swish（hswish）和mish激活功能。
此外，添加了基于cuda的一键式功能（支持标签平滑）。
新添加一个“指数移动平均线（EMA）”运算符。
添加卷积运算，例如coord-conv2d和dynamic-conv2d（dy-conv2d）。
一些运算符是使用pytorch cuda扩展实现的，因此您需要先对其进行编译：
$ python setup.py
【文件预览】：
pytorch-loss-master
----amsoftmax.py(2KB)
----dual_focal_loss.py(1KB)
----triplet_loss.py(1024B)
----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(4KB)
--------dual_focal_loss.py(1KB)
--------triplet_loss.py(1024B)
--------hswish.py(4KB)
--------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)
--------__init__.py(1KB)
--------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)
----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)
--------soft_dice_kernel_v2.cu(16KB)
--------soft_dice_kernel.cu(10KB)
--------common.hpp(1KB)
--------ohem_label_kernel.cu(9KB)
--------lovasz_softmax.cu(28KB)
----generalized_iou_loss.py(1KB)
----ema.py(3KB)
----pc_softmax.py(6KB)