文件名称:Loss_ToolBox-PyTorch:PyTorch实现焦点损失和Lovasz-Softmax损失
文件大小:252KB
文件格式:ZIP
更新时间:2024-06-12 22:01:20
computer-vision deep-learning pytorch pytorch-implmention focal-loss
Loss_ToolBox 介绍 该存储库包括3D图像分割的几处损失。 (PS:从借一些代码) (根据常规实现修改)
【文件预览】:
Loss_ToolBox-PyTorch-master
----TverskyLoss()
--------__init__.py(0B)
--------multitverskyloss.py(2KB)
--------binarytverskyloss.py(5KB)
--------tversky_loss_test.py(1KB)
----utils()
--------utils.py(461B)
--------__init__.py(0B)
----loss_demo.ipynb(170KB)
----DiceLoss()
--------dice_loss.py(9KB)
--------__init__.py(0B)
----LICENSE(1KB)
----demo.py(3KB)
----LovaszSoftmax()
--------lovasz_loss_demo.ipynb(173KB)
--------lovasz_loss.py(3KB)
----README.md(440B)
----FocalLoss()
--------focal_loss.py(6KB)
--------__init__.py(0B)
--------FocalLoss_test.py(2KB)
--------README.md(492B)
----.gitignore(1KB)