eriklindernoren / pytorch-deep-dream Goto Github PK

Currently the code fails if using Cuda. Line 24 should read

- image.data += norm_lr * image.grad.data
+ image.data += Tensor([norm_lr]) * image.grad.data

On lines 17 and 73, torch.cuda.is_available should instead be torch.cuda.is_available().

err

When I run this: python deep_dream.py --input_image images/supermarket.jpg
I get the following error"
invalid value encountered in true_divide
zoom = (numpy.array(input.shape) - 1) / zoom_div
/System/Library/Frameworks/Python.framework/Versions/2.7/Extras/lib/python/scipy/ndimage/interpolation.py:532: UserWarning: From scipy 0.13.0, the output shape of zoom() is calculated with round() instead of int() - for these inputs the size of the returned array has changed.
"the returned array has changed.", UserWarning)
Dreaming: 10%|█ | 1/10 [00:01<00:11, 1.23s/it]
Traceback (most recent call last):
File "deep_dream.py", line 86, in
num_octaves=args.num_octaves,
File "deep_dream.py", line 51, in deep_dream
input_image = octave_base + detail
ValueError: operands could not be broadcast together with shapes (1,3,31,44) (1,3,22,31)