参考链接:
windows10下安装TensorFlow Object Detection APIblog.csdn.net具体安装步骤看原链接就好,这里记录一下安装中遇到的问题:
- No module named tensorflow.compat.v1
安装完成后,执行下面的验证代码,出现报错No module named tensorflow.compat.v1
python object_detection/builders/model_builder_test.py
tensorflow.compat.v1 这个包只会出现在 tf2.0 版本中,而我的环境是 tf1.14
说明在第一步:
git clone https://github.com/tensorflow/models.git
代码拉下来之后,要 checkout 到相应版本分支去:
git checkout remotes/origin/r1.13.0
如果你不清楚相应的版本有哪些,可以用下面的命令查看一下:
git brach -a
我的 tensorflow 版本是 1.14,那我在这选 1.13.0 的分支就可以了。
然后把流程再走一遍即可。
重新安装完成,再执行验证代码
python object_detection/builders/model_builder_test.py
出现了新的报错:
- module 'google.protobuf.descriptor' has no attribute '_internal_create_key'
这里 给出了解释:
查看一下我自己的版本:
一个是 3.9.0 一个是3.12.3,差得有点多
升级一下版本就可以了:
pip install --upgrade protobuf
再一次执行验证代码:
python object_detection/builders/model_builder_test.py
这次终于顺利执行了:
说明 object_detection API 终于成功安装好了!