machine-learning - Tensorflow:没有属性 'load_labelmap'

Question

我按照 Github 上描述的步骤安装 Object Detection API然后我运行这个脚本:

python object_detection/builders/model_builder_test.py

并且测试成功，所以我认为一切都设置正确。接下来，我尝试使用 qtconsole 运行 Jupyter Notebook 来检测测试图像中的对象。但是它返回此错误:

AttributeError                            Traceback (most recent call last)
<ipython-input-3-be6fe1ba8733> in <module>()
----> 1 from utils import label_map_util
      2 
      3 from utils import visualization_utils as vis_util
      4 

~\Desktop\Objectdetection\models-master\object_detection\utils\label_map_util.py in <module>()
     20 import tensorflow as tf
     21 from google.protobuf import text_format
---> 22 from object_detection.protos import string_int_label_map_pb2
     23 
     24 

~\Desktop\Objectdetection\models-master\object_detection\object_detection.py in <module>()
    114 
    115 
--> 116 label_map = label_map_util.load_labelmap(PATH_TO_LABELS)
    117 categories = label_map_util.convert_label_map_to_categories(label_map, max_num_classes=NUM_CLASSES, use_display_name=True)
    118 category_index = label_map_util.create_category_index(categories)

AttributeError: module 'utils.label_map_util' has no attribute 'load_labelmap'

有人知道这个问题的原因是什么吗？

谢谢。

Answer 1

在文件~\Desktop\ObjectDetection\models-master\object_detection\utils\label_map_util.py

尝试更改此设置:

从 object_detection.protos 导入 string_int_label_map_pb2

对此:

从 protos 导入 string_int_label_map_pb2

说明:

模块 label_map_util 中的函数 load_labelmap 无法访问，因为 string_int_label_map_pb2 导入失败。

如果查看 print(dir(label_map_util)) 的输出，您就可以看到这一点。

使用object_detection.protos时:

['__builtins__'、'__cached__'、'__doc__'、'__file__'、'__loader__'、'__name__'、'__package__'、'__spec__'、'logging'、'text_format'、'tf' ]

更改 protos 的相对路径后，该函数应该可以访问:

['__builtins__'、'__cached__'、'__doc__'、'__file__'、'__loader__'、'__name__'、'__package__'、'__spec__'、'_validate_label_map'、'convert_label_map_to_categories'、'create_category_index' , 'get_label_map_dict', 'load_labelmap', 'logging', 'string_int_label_map_pb2', 'text_format', 'tf']