Open iaroslav-ai opened 8 years ago
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
I used now
test = ArrayIterator(np.random.rand(256, 3*224*224), lshape=(3, 224, 224))
I also added in the end
print model.get_outputs(test)
It seems that now neon does something for some time, but then in the get outputs function it gives the error
Traceback (most recent call last):
File "/home/iaroslav/neon-examples/googlenet/demo.py", line 104, in <module>
print model.get_outputs(test)
File "/usr/local/lib/python2.7/dist-packages/neon/models/model.py", line 257, in get_outputs
(dim0, dim1) = x.shape
AttributeError: 'list' object has no attribute 'shape'
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.