search

xunhuang1995 / AdaIN-style

Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization
https://arxiv.org/abs/1703.06868
MIT License
1.47k stars 192 forks source link

Color saturation? #37

Open litingfeng opened 5 years ago

litingfeng commented 5 years ago

Not sure if I should call the problem as color saturation. In my generated images, I found out there's some pixel chunks, like the black area in the image below.

im

How should I solve this problem? FYI, the original images are 3232, I resized them to 256256, then train the decoder.

Thanks,

ZhongliangGuo commented 9 months ago

just do torch.clamp(tensor, 0, 1)