Caffe: AttributeError: 'module' object has no attribute 'LabelMap'

[xizi]

When I perform command ./data/VOC0712/create_data.sh, it report
Traceback (most recent call last):
File "/root/lishengxi/caffe/data/VOC0712/../../scripts/create_annoset.py", line 103, in
label_map = caffe_pb2.LabelMap()
AttributeError: 'module' object has no attribute 'LabelMap' error

[weiliu89]

I am guessing you didn't setup the PYTHONPATH environment variable correctly.

You can either try:

export PYTHONPATH=$CAFFE_ROOT/python:$PYTHONPATH

where CAFFE_ROOT is the directory you have the code

or put it in your ~/.bashrc and run:

source ~/.bashrc

[xizi]

Wei, thanks a lot.
I solve the problem according to your solution.

[dizcza]

Hi, I'm still getting this error though I've already added $CAFFE_ROOT/python to my PYTHONPATH.
But I've built caffe from a master tree rather than from ssd tree. Might it be the case why I'm getting AttributeError: 'module' object has no attribute 'LabelMap' error, in other words, does master's and ssd's caffe installations differ?

[weiliu89]

You have to build the ssd branch.

[cherishtttz]

Hi, I still get this problem though I have already added $CAFFE_ROOT/python to my PYTHONPATH and updated the ssd branch then rebuilt it just a few days ago. I named the ssd branch 'caffe-ssd' .

lab@lab:~/caffe-ssd$ echo $PYTHONPATH
:/home/lab/py-faster-rcnn/lib:/home/lab/py-faster-rcnn/caffe-fast-rcnn/python:/home/lab/caffe-ssd/python

[FumingX]

I did:
make py
make test -j8
make runtest -j8
and my ~/.bashrc:
export PATH=$PATH:/usr/local/cuda-8.0/bin export LD_LIBRARY_PATH=:/usr/local/cuda-8.0/lib64 export PYTHONPATH=/home/fuming/caffe/python:$PYTHONPATH
still got
'module' object has no attribute 'LabelMap'

[wishinger-li]

@FumingX you solved this?
i got this problem now,can you give me some advice?

[FumingX]

@wishinger-li I just restarted the computer after installation, and it worked.

[jackyfriend]

@FumingX I have solved this problem following these steps:

1. export PYTHONPATH=$CAFFE_ROOT/python:$PYTHONPATH
2. make py
3. run ./data/VOC0712/create_data.sh
   hope that may help you.

[PiseyYou]

1, open this file, ~/caffe-ssd/scripts/create_annoset.py
2,edit the file in path between [import sys] and [from caffe.proto import caffe_pb2].add this
sys.path.insert(0,'/home/xxx/caffe_ssd/python’)
beside add the /caffe-ssd/python to the bashrc way,this way work for me, hope it can help others.

[jamessmith123456]

sys.path.insert(0,'/home/xxx/caffe_ssd/python’) . This works for me while "adding to .bashrc" not.