afruehstueck / tileGAN

Code for TileGAN: Synthesis of Large-Scale Non-Homogeneous Textures (SIGGRAPH 2019)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

To those whom cannot reproduce the results on own datasets

hubert0527 opened this issue · comments

TL;DR The author has modified the PGGAN architecture in their project, so your model pretrained on the original PGGAN codebase cannot be used here. You can delete it now.

Related Issue:
#10

Observation
Since the network architecture is altered, so it is expected to generate random pixels like this:
image

You may wonder why the preview thumbnails are working as expected. It is because the preview thumbnails are generated with the graphs stored in your own checkpoint, while the canvas above is generated with an altered model network architecture that breaks your model.

Solution

  1. Replace these lines with these lines
  2. Train the PGGAN model all over again.
  3. Replace this line with the codes below. This avoids loading modules that do not exist.
if out_res < resolution_log2:
    images_out = tf.identity(x)
else:
    images_out = torgb(x, res)
  1. You're good to go.