search

marcbelmont / cnn-watermark-removal

Fully convolutional deep neural network to remove transparent overlays from images
1.24k stars 228 forks source link

Where to place training Data? #16

Closed Skarlett closed 6 years ago

Skarlett commented 6 years ago

I'm not exactly sure where to place training data, if it should be in a specific format or not. Error message: Source images are missing!

Please document these into the README.md file. Its quite stressful to try and find.

heliocarbex commented 6 years ago

I think that this code is not updated. At the file dataset.py, we have in line 138: def get_images(): data = None for path in glob('cifar-10-batches-py/data_batch_*'):

So, this code is expecting a folder named "cifar-10-batches-py" and files matching that string "databatch*".