How to improve the mask-shift results

[gsdndeer]

Hi authors, thanks for releasing this great work!

I am trying to use mask-shift technique on my image-to-image translation task in medical images. I trained my diffusion model with crop data augmentation. But I can’t get the smooth results like yours. Is there any way to improve it?

The following are the results of my task.

This is the generated result of the first patch.

Then, I shift the patch right and leave a half of overlapped regions with the previous patch.

This is the result of the step 2.