You can find Dual Focal Loss implementation in dual_focal_loss.py
from dual_focal_loss import DualFocalLoss
criterion = DualFocalLoss()
output = model(data)
loss = criterion(logits, targets)
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}
}