Skip to content

Commit

Permalink
Add stereo matching losses (#6554)
Browse files Browse the repository at this point in the history
* 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: Joao Gomes <jdsgomes@fb.com>
  • Loading branch information
TeodorPoncu and jdsgomes authored Sep 21, 2022
1 parent 56e707b commit 2c1022e
Showing 1 changed file with 503 additions and 0 deletions.
Loading

0 comments on commit 2c1022e

Please sign in to comment.