tensorflow的object_detection安装

安装环境tensorflow2.5+python3.8.8

https://blog.csdn.net/xubuhui/article/details/100915320  

1、下载模型源码

下载地址:https://github.com/tensorflow/models

2、安装protoc

地址:https://github.com/protocolbuffers/protobuf/releases

1)选择win64版本安装,解压,将bin路径设为环境变量

2)在cmd命令行测试命令:protoc

3、在anaconda里面添加环境变量

(1)找到Anaconda路径,例如:D:\Programs\Anaconda3\Lib\site-packages

(2)在该文件夹下新建一个txt文件,保存为tensorflow_model.pth;内容为刚才下载的模型源码路径

F:\component\HandDetection\models-master\research
F:\component\HandDetection\models-master\research\slim

3、编译安装object_detection

1)编译proto文件

(1)方法1:在models/research下运行windows powershell,输入如下命令:(下面的ok)

  https://blog.csdn.net/qq_28019591/article/details/82023949

Get-ChildItem object_detection/protos/*.proto | Resolve-Path -Relative | %{ protoc $_ --python_out=. }

运行完成后,可以检查object_detection/protos/文件夹,如果每个proto文件都成了对应的以py为后缀的python源码,就说明编译成功了。

(2)方法2:在Anaconda Prompt这个窗口,切换到你下载的TensorFlow模型存储库的research目录下,根据实际情况更改

  https://zhuanlan.zhihu.com/p/347860796

cd D:\models\research 
protoc object_detection/protos/*.proto --python_out=.

编译完你可以在object_detection\protos目录下看到每个proto文件变成py文件

2)编译安装object_detection

  https://zhuanlan.zhihu.com/p/347860796

cd object_detection/packages/tf2 
python -m pip install .

4、测试object_detection的api

cd research\object_detection\colab_tutorials
jupyter notebook
找到object_detection_tutorial.ipynb文件打开
cell->run all(这里不用再编译protoc文件了,我是单步执行了python代码部分,bash部分没执行)

 

posted @ 2021-07-28 19:03  kuaqi  阅读(357)  评论(0编辑  收藏  举报