red color bleeding

[juneleung]

The model seems to be particularly sensitive to red areas, such as lips, red feathers, pants. The output red will have colour bleeding. How to fix this problem? Thanks!

input

reference is a black and white photo.
output

[maria-larchenko]

Hi @juneleung!

Yes, true, I also noticed it. I attribute this problem to the dataset, that contains not that many images with red shades. These is no easy fix for this, unfortunately. I'll try to prepare and publish our data, then it'll be possible to improve the quality