• 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
presets.py 5.08 KB