/stnm.pytorch

Pytorch code for spatial transformer network with mask

Primary LanguageJupyter NotebookMIT LicenseMIT

Stargazers