search

beala / deep-image-prior-tensorflow

An implementation of https://dmitryulyanov.github.io/deep_image_prior for tensorflow.
Other
68 stars 19 forks source link

Small bug in load_image #3

Open z-fabian opened 6 years ago

z-fabian commented 6 years ago

You might want to change 

def load_image(filename, dim):
    with open(image_name, 'rb') as f:
raw_image = tf.image.decode_png(f.read())
...

to

def load_image(filename, dim):
    with open(filename, 'rb') as f:
raw_image = tf.image.decode_png(f.read())
...