Skip to content

Linwei94/DualFocalLoss

Repository files navigation

Dual Focal Loss for Calibration

You can find Dual Focal Loss implementation in dual_focal_loss.py

Usage

from dual_focal_loss import DualFocalLoss

criterion = DualFocalLoss()

output = model(data)
loss = criterion(logits, targets)

Citation

If you find this repo useful, please cite our paper.

@misc{tao2023dual,
      title={Dual Focal Loss for Calibration}, 
      author={Linwei Tao and Minjing Dong and Chang Xu},
      year={2023},
      eprint={2305.13665},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

About

Code for "Dual Focal Loss for Calibration" (ICML 2023)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages