search

marcbelmont / cnn-watermark-removal

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

error when try to run python watermarks.py --image assets/cat.png --selection assets/cat-selection.png #12

Closed aalzooke closed 6 years ago

aalzooke commented 6 years ago

I followed your requirement but when i try to run python watermarks.py --image assets/cat.png --selection assets/cat-selection.png i got the following error. screenshot from 2018-06-05 16-05-12

marcbelmont commented 6 years ago

@aalzooke which version of Tensorflow are you using? I built this using tensorflow==1.3.0. In recent version Dataset is not in contrib. You can try tf.data.Dataset... instead.