/CS238-VirtualReality

Final project for SJTU CS238

Parametrized-brushstroke Image Style Transfer

This is a Pytorch implementation of the paper "Rethinking Style Transfer: From Pixels to Parameterized Brushstrokes".

Prerequisites

Pytorch (>= 1.8 (required))

Pytorch cluster

Neural monitor (For logging)

Running the code

python main.py /path/to/content/image /path/to/style/image

Type python main.py to see all available options.

Similarities/differences to the official implementation

Similarities

  • The renderer and the initialization schemes are no-brainer adaptations from the official repo.
  • The training schemes (including learning rates and optimizers) are the same.
  • The visual results look largely similar.

Differences

  • Some minor details regarding the style losses are different.

References

Official Tensorflow implementation

VGG pretrained weight file