search

pystiche / papers

Reference implementation and replication of prominent NST papers
BSD 3-Clause "New" or "Revised" License
4 stars 1 forks source link

Add author info replication script in gatys_ecker_betghe_2016 #296

Closed jbueltemeier closed 1 year ago

jbueltemeier commented 1 year ago

The replication of gatys_ecker_betghe_2016 shows clear differences in the results. For this reason, I have contacted the author. He provided helpful tips for improving the results. This information on the hyperparameters is integrated with this PR. The most important changes are:

  1. the use of the activations after the ReLU function.
  2. the use of layer_weights depending on the number of channels.
  3. random initialisation of the image and more optimisation steps.

Due to the random initalisation, an exact replication is no longer possible, but the results show a clear improvement and a more similar stylisation to the original images.