-
Notifications
You must be signed in to change notification settings - Fork 68
Error while trying to get outputs of googlenet #5
Comments
The ArrayIterator needs to have the data as a 2D matrix with the dim (N, CHW). So the CHW part needs to be squashed and it needs the shape parameter set to (C, H, W). Try changing line test = ArrayIterator(np.random.rand(256,224,224,3)) to test = ArrayIterator(np.random.rand(256, 3_224_224), lshape=(3, 224, 224)) |
Ah I see
Did I do something wrong again? :) I have my updated code in here Thanks for your response! :D |
Could you pull the most recent model zoo repo and try again, I think the model needed to be updated to use the SingleOutputTree |
Thanks for your response! I will try to pull the most recent model zoo. |
I guess this can be closed due to #3 ? |
Not sure if this relates to #3. Would be good to make sure that this is working for you. |
I try to get output of googlenet with some randomly generated inputs using a modified version of googlenet_neon.py. I get warning and error messages below. Did I made some mistake or is there a bug? The code that I use is below error messages.
Messages:
The code:
https://gist.github.com/iaroslav-ai/949ee0200152236937c4d0b29f70f70e
Assume that googlenet model is located at /home/iaroslav/temp/googlenet.p.
The text was updated successfully, but these errors were encountered: