inpainting_gmcnn icon indicating copy to clipboard operation
inpainting_gmcnn copied to clipboard

Train on my own dataset

Open jinyu121 opened this issue 4 years ago • 3 comments

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)

image

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)

jinyu121 avatar Jan 16 '20 13:01 jinyu121

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?

shepnerd avatar Jan 17 '20 03:01 shepnerd

Stage 1:

image

image

image

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

image

Tensorboard record of Stage2 crashed😂

Seems I should wait for longer at Stage 1?

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

jinyu121 avatar Jan 17 '20 03:01 jinyu121

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.

shepnerd avatar Jan 17 '20 06:01 shepnerd