如何解决错误 "cannot import name 'string_int_label_map_pb2' from 'object_detection.protos'"

How to resolve the error "cannot import name 'string_int_label_map_pb2' from 'object_detection.protos'"

我和我的朋友们正在参加黑客马拉松,并且被困在这个关于训练对象检测模型的教程中:

https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html

但是在三个人尝试严格按照上面的教程操作之后,我们都卡在了创建 TfRecord 的步骤上:

https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html#create-tensorflow-records

我们收到错误消息:

ImportError: cannot import name 'string_int_label_map_pb2' from 'object_detection.protos'

我们是三个不同的人,使用三台不同的机器,得到了这个结果,总共花费了 20 多个小时的时间在谷歌上搜索、重试、牺牲一些东西给可怕的神,但没有什么能让我们越过这个障碍。

是否有更好的方法来实现结果或解决问题?我们对 Tensorflow 没有经验,只是想用它来训练一个可以转换为将被 ml.net.

使用的 onnx 模型的模型

感谢 @EdwinCheong 为我们指明了正确的方向。

在遵循此链接教程之前,基本上通过在 Visual Studio 安装程序中下载 Windows 10 SDK 和一堆与 c++ 相关的包来解决问题: https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/install.html#tf-models-install 与安装对象检测相关。不管教程怎么说,似乎都需要 CUDA