Не могу загрузить метки классов уже обученной модели в tf

Использую готовую нейросетевую модель Faster_rcnn_inception_resnet_v2_atrous_coco_2018_01_28 в области Object Detection. Загружаю метки классов для разграничения изображений на кластеры:

label_map = label_map_util.load_labelmap('/models_master/research/object_detection/data/mscoco_label_map.pbtxt')
categories = label_map_util.convert_label_map_to_categories(label_map, max_num_classes=90, use_display_name=True)
category_index = label_map_util.create_category_index(categories)

Получаю ошибку:

~\AppData\Roaming\Python\Python38\site-packages\object_detection\utils\label_map_util.py in load_labelmap(path)
    130   """
    131   with tf.gfile.GFile(path, 'r') as fid:
--> 132     label_map_string = fid.read()
    133     label_map = string_int_label_map_pb2.StringIntLabelMap()
    134     try:

AttributeError: module 'tensorflow' has no attribute 'gfile'

В официальной документации к TensorFlow прочитал, что метод был перенесён в tf._api.v2.gfile.GFile. Я поменял в исходном коде label_map_util.py на tf._api.v2.gfile.GFile, но tf всё равно ссылается на gfile:

~\AppData\Roaming\Python\Python38\site-packages\object_detection\utils\label_map_util.py in load_labelmap(path)
    130   """
    131   with tf._api.v2.gfile.GFile(path, 'r') as fid:
--> 132     label_map_string = fid.read()
    133     label_map = string_int_label_map_pb2.StringIntLabelMap()
    134     try:

AttributeError: module 'tensorflow' has no attribute 'gfile'

Помогите решить проблему, либо подскажите, как ещё можно загрузить готовые метки классов.