search

jrosebr1 / bat-country

A lightweight, extendible, easy to use Python package for deep dreaming and image generation with Caffe and CNNs.
MIT License
246 stars 62 forks source link

Error running visualize_layers.py #10

Closed HandsomeDevilv112 closed 3 years ago

HandsomeDevilv112 commented 3 years ago

I'm trying to replicate the results from https://www.pyimagesearch.com/2015/08/03/deep-dream-visualizing-every-layer-of-googlenet/ However, when I get near the end, it crashes, and I'm not quite sure what to do with it. Bat Country can run single images without difficulty, but doesn't quite make it here. To note, I am using the bvlc_googlenet model that I got when looking into deepdreamanim. my command line is: python visualize_layers.py --base-model /root/deepdream/caffe/models/bvlc_googlenet --image g1.jpg --output bvlc_googlenet

The crash looks like:

[INFO] processing layer inception_5b/pool 145/152 [INFO] processing layer inception_5b/pool_proj 146/152 [INFO] processing layer inception_5b/relu_pool_proj 147/152 [ERROR] cannot use layer inception_5b/relu_pool_proj [INFO] processing layer inception_5b/output 148/152 [INFO] processing layer pool5/7x7_s1 149/152 F0621 18:21:13.746654 2164 blob.cpp:32] Check failed: shape[i] >= 0 (-3 vs. 0) Check failure stack trace: Aborted (core dumped)

I'd appreciate any and all help. Thank you.

HandsomeDevilv112 commented 3 years ago

rebooted the machine, used the "clouds" image included in the repo, and have been able to successfully run visualize_layers.py Closing.