Based on your work, I found this nice implementation of DeepFill v2 for PyTorch, I thought it might be useful to others: https://github.com/nipponjo/deepfillv2-pytorch