About the input of discriminator

[blessingforever]

Thank you for sharing this great work!
I have a question about the code. According to your paper, the input of discriminator should include both the segmentation maps and the raw image. But in your code, It seems like only the segmentation maps are used as input. So where the raw image is used?Or maybe I have some misunderstanding about that