karpathy / neuraltalk

NeuralTalk is a Python+numpy project for learning Multimodal Recurrent Neural Networks that describe images with sentences.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

change in extract_features needed due to caffe update

samim23 opened this issue · comments

commented

This code line:
features = out[net.outputs[0]].squeeze(axis=(2,3))
Has to be this in order to work with the newest caffe:
features = out[net.outputs[0]]

Additionally remove this line:
caffe.set_phase_test()

+1 I had to modify extract_features.py with these changes to get it to run on latest caffe