search

marcbelmont / cnn-watermark-removal

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

Little bug reminder #4

Closed huhaoranak47 closed 6 years ago

huhaoranak47 commented 6 years ago

1 watermarks.py line 12 should be tf.flags.DEFINE_string('logdir', '/log', 'Log directory') 2 you should build an directory named 'data' and put the traindata in that 3 training cost lots of time

huhaoranak47 commented 6 years ago

4 watermarks.py line 7 should be import matplotlib matplotlib.use('Agg') import matplotlib.pyplot as plt

marcbelmont commented 6 years ago

I would be happy to merge your pull request.

huhaoranak47 commented 6 years ago

Thanks a lot man I will be appreciated that if you share the model.ckpt file it cost almost 2 days to train the model

marcbelmont commented 6 years ago

Pretrained weights have been provided in https://github.com/marcbelmont/cnn-watermark-removal/issues/5