zlmzju / part_reid

Code for ICCV2017 paper: Deeply-Learned Part-Aligned Representations for Person Re-Identification
123 stars 50 forks source link

KeyError: 'normed_feature' not found #10

Closed rohanchandra30 closed 6 years ago

rohanchandra30 commented 6 years ago

In the extract_features method in the Extractor() class, there is a line:

normed_features=self.net.blobs['normed_feature'].data.copy()

I checked the _caffe.cpp file in ./caffe/python/caffe/_caffe.cpp and I then checked all the properties for net.blobs and 'normed_feature' was not one of them.

Any ideas on how to resolve this?

Thanks

zlmzju commented 6 years ago

The key of normed_feature is the blob name defined in network architecture. You can find it in 'models/demo/test.prototxt'.

zlmzju commented 6 years ago

The key of normed_feature is the blob name defined in network architecture. You can find it in 'models/demo/test.prototxt'.