change in extract_features needed due to caffe update

karpathy / neuraltalk

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

5.4k stars 1.32k forks source link

change in extract_features needed due to caffe update #31

Open samim23 opened 8 years ago

samim23 commented 8 years ago

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()

alyxb commented 8 years ago

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