training with my own dataset

[Hunterwolf88]

Hello there,
first of all congratulations for this awesome project!

I need help regarding the training process.
I learned from #15 that we need a training and a validation dataset but I still have some confusion.

Is the training dataset just a set of pictures with white rectangles covering some parts and the validation dataset the same pictures without any occlusion?

[JiahuiYu]

No. The training dataset and validation dataset are complete images without masks. The code here will sample a random mask to images for inpainting.

[Hunterwolf88]

thank you so much