Skip to content
A pytorch port of google-research/google-research/robust_loss/
Branch: master
Clone or download
Latest commit 3f91314 Jun 16, 2019

README.md

A General and Adaptive Robust Loss Function

This directory contains reference code for the paper A General and Adaptive Robust Loss Function, Jonathan T. Barron CVPR, 2019

The code is implemented in Pytorch, and is a port of the TensorFlow implementation at: https://github.com/google-research/google-research/tree/master/robust_loss. The required packages are listed in requirements.txt.

To use this code, include general.py or adaptive.py and call the loss function. general.py implements the "general" form of the loss, which assumes you are prepared to set and tune hyperparameters yourself, and adaptive.py implements the "adaptive" form of the loss, which tries to adapt the hyperparameters automatically and also includes support for imposing losses in different image representations. The probability distribution underneath the adaptive loss is implemented in distribution.py.

If you use this code, please cite it:

@article{BarronCVPR2019,
  Author = {Jonathan T. Barron},
  Title = {A General and Adaptive Robust Loss Function},
  Journal = {CVPR},
  Year = {2019}
}
You can’t perform that action at this time.