小编Dhe*_*oho的帖子

我尝试在 Tensorflow 2 上运行 Tensorflow 对象检测 API，但收到该错误，有人可以提供解决方案吗？

代码 ：

装载机

def load_model(model_name):\n  base_url = \'http://download.tensorflow.org/models/object_detection/\'\n  model_file = model_name + \'.tar.gz\'\n  model_dir = tf.keras.utils.get_file(\n    fname=model_name, \n    origin=base_url + model_file,\n    untar=True)\n\xe2\x80\x8b\n  model_dir = pathlib.Path(model_dir)/"saved_model"\n\xe2\x80\x8b\n  model = tf.saved_model.load(str(model_dir))\n  model = model.signatures[\'serving_default\']\n\xe2\x80\x8b\n  return model\n

加载标签图

标签将索引映射到类别名称，因此当我们的卷积网络预测为 5 时，我们知道这对应于飞机。这里我们使用内部实用函数，但是任何返回将整数映射到适当字符串标签的字典的东西都可以

# List of the strings that is used to add correct label for each box.\nPATH_TO_LABELS = \'data/mscoco_label_map.pbtxt\'\ncategory_index = label_map_util.create_category_index_from_labelmap(PATH_TO_LABELS, use_display_name=True)\n

为了简单起见，我们将在 2 个图像上进行测试：

# If you want to …