MathiasGruber / PConv-Keras

Unofficial implementation of "Image Inpainting for Irregular Holes Using Partial Convolutions". Try at: www.fixmyphoto.ai
MIT License
903 stars 253 forks source link

Quick code implementation while receive data problem #60

Open hiterjoshua opened 4 years ago

hiterjoshua commented 4 years ago

Thanks for your awesome reproduce work about partial convolution work, while I made two quick implementation of this repo on two different machines, both of my implementation seem to have some problems about data, specifically, I suspect that the problem caused by flow_from_directory function.

  1. The first error sceenshot looks like: image it says the fit_generator receives empty batch.

  2. its really awkward since I didn't change the code, I just provide the training data path. and the training data organizes like this: train/ CelebHQ/ 001.jpg 002.jpg ... Any suggestions or discussiones are welcomed!

burhr2 commented 3 years ago

Hi maybe you can try with the notebook and see