Imported from https://github.com/EmptySamurai/pytorch-reconet