naoto0804 / pytorch-AdaIN

Unofficial pytorch implementation of 'Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization' [Huang+, ICCV2017]
MIT License
1.08k stars 208 forks source link

input size changed #23

Closed MarStarck closed 5 years ago

MarStarck commented 5 years ago

when I input a content image with size 414 274, the output size turns out to be 776 512, and it becomes very blur. I guess the up-sample module may have some problems

MarStarck commented 5 years ago

sorry, It just need to set content_size and style_size to be 0 to keep original size, otherwise will set to 512 by default.