/torchfields

A PyTorch add-on for working with image mappings and displacement fields, including Spatial Transformers

Primary LanguagePythonMIT LicenseMIT