Initial vgg_normalized.pth, decoder.pth

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

Initial vgg_normalized.pth, decoder.pth #33

Closed Nadavc220 closed 3 years ago

Nadavc220 commented 3 years ago

Hi, Can you explain what are the initial decoder, vgg weights given? I am referring the weights which are instructed to be set in ./models before training.

Thanks.

naoto0804 commented 3 years ago

decoder.pth is pre-trained decoder (provided in the original implementation) for testing purpose only. If you train the model. you can use the trained decoder instead.