Imported from https://github.com/dongheehand/MemoPainter-PyTorch