LabelMap в Python Caffe

Из модуля caffe импортирую класс LabelMap таким образом:

from caffe.proto.caffe_pb2 import LabelMap

Но получаю ошибку:

ImportError: cannot import name 'LabelMap'

Подскажите, пожалуйста, где расположен класс LabelMap в caffe. Может он находится в другом файле? Или может в текущей версии caffe название класса изменилось?

Если что, caffe устанавливал таким образом:

sudo apt install caffe-cuda