Train on my own dataset

[jinyu121]

Hi~

I am training on my own dataset. In stage 2, I get this image(up to down: GT, image feed to the generator, result)

As you can see, a darker mask is still on the image.

Have you ever experienced that? Or, what should I do next? Wait for a longer time, or optimize some hyper-parameters? (Now the parameters are copied from your code and not changed)

[shepnerd]

The posted results did seem problematic. Did the first stage training converge? Could you share the training loss trend of both first and second stage?

[jinyu121]

Stage 1:

At last, Stage 1 can get images like this(Up to down: gt, input, output):

Tensorboard record of Stage2 crashed😂

Seems I should wait for longer at Stage 1?

( 1st time to try GAN... May have tons of questions😂)

[shepnerd]

It seems you need to training the first stage longer since you conduct natural scenes inpainting. Besides, you may reverse the channel order of the shown images. It appears in BRG instead of RGB channels.