A question about masks in image repair tasks

Thank you for this incredible paper and repository, I have a question:
I want to run an experiment against an image restoration task, and I want to change the centre mask to my own mask image. 
An answer would be appreciated.