• Ponku's avatar
    Add stereo matching losses (#6554) · 2c1022e3
    Ponku authored
    
    
    * Moved more losses into classes
    
    * Added photometric loss
    
    * quick fix for ssim loss return value
    
    * added references
    
    * replaced with unsqueeze
    
    * renaming variables
    
    * add ref to consistency loss
    
    * made mask optional everywhere. generalised photometric displacement
    
    * smoothness typo
    
    * fixed flow channel selection bug
    
    * aligned with training script
    Co-authored-by: default avatarJoao Gomes <jdsgomes@fb.com>
    2c1022e3
losses.py 17.1 KB