Skip to content

SJTU-Lucy/CS238-VirtualReality

Repository files navigation

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

About

Final project for SJTU CS238

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published