Baugh, M. PIE-torch [Computer software]. https://github.com/matt-baugh/pytorch-poisson-image-editing