1. 22 Sep, 2022 1 commit
    • Ponku's avatar
      Add stereo preset transforms (#6549) · 0fcfaa13
      Ponku authored
      
      
      * Added transforms for Stereo Matching
      
      * changed implicit Y scaling to 0.
      
      * Adressed some comments
      
      * addressed type hint
      
      * Added interpolation random interpolation strategy
      
      * Aligned crop get params
      
      * fixed bug in RandomErase
      
      * Adressed scaling and typos
      
      * Adressed occlusion typo
      
      * Changed parameter order in F.erase
      
      * fixed random erase
      
      * Added inference preset transform for stereo matching
      
      * added contiguous reshape to output tensors
      
      * Adressed comments
      
      * Modified the transform preset to use Tuple[int, int]
      
      * adressed NITs
      
      * added grayscale transform, align resize -> mask
      
      * changed max disparity default behaviour
      
      * added fixed resize, changed masking in sparse flow masking
      
      * update to align with argparse
      
      * changed default mask in asymetric pairs
      
      * moved grayscale order
      
      * changed grayscale api to accept to tensor variant
      
      * mypy fix
      
      * changed resize specs
      
      * adressed nits
      
      * added type hints
      
      * mypy fix
      
      * mypy fix
      
      * mypy fix
      Co-authored-by: default avatarJoao Gomes <jdsgomes@fb.com>
      0fcfaa13
  2. 21 Sep, 2022 1 commit
    • 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