tensorflow 的 models 模块非常有用,不仅实现了各种模型,也包括了 原作者 训练好的模型及其使用方法,本文 以 object detection 为例 来说明如何使用 训练好 的模型;
首先呢,还是建议 去 官网 看看使用方法,因为 tensorflow 的版本混乱,网上教程针对的版本各不相同,所以各种坑;
下面是正题,本文针对 windows 操作系统;
第一步:下载 models 模块,解压
https://github.com/tensorflow/models
第二步:安装 protoc
https://github.com/protocolbuffers/protobuf/releases 从这里下载,选择适合自己的版本;
下载后复制到 models 所在的文件夹下,解压,生成 bin、include;
将 bin 下的 protoc.exe 复制到 C:\Windows\System32 文件夹下;
cmd 运行 protoc,出现如下界面,说明安装成功;
第三步:编译 protoc
在 models/research 下运行 Windows PowerShell 【运行 PowerShell】
运行命令
protoc object_detection/protos/*.proto --python_out=.
运行完成后,检查 object_detection/protos 文件夹,如果每个 proto 文件都变成了 py 文件,表示编译成功
第四步:添加环境变量
添加这两个目录
...\models\research ...\models\research\slim
至于怎么添加,你可以用常规的设置 环境变量的方式,官方是 PYTHONPATH;
网上有 添加 .pth 文件,我实验未成功;
第五步:测试 API 是否安装成功
python object_detection/builders/model_builder_test.py
出现上图表示成功;
第六步:执行已经训练好的模型
执行 object_detection/object_detection_tutorial.ipynb 文件 【执行方法 我的博客】
参考资料:
https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/installation.md 官网
https://www.jianshu.com/p/6f3ea0d82fae 物体检测TensorFlow Object Detection API (一)安装
https://www.jb51.net/article/162968.htm windows10下安装TensorFlow Object Detection API的步骤
https://www.cnblogs.com/2dogslife/p/10264325.html Tensorflow Object Detection API 安装
https://blog.csdn.net/qq_38593211/article/details/82822162 TensorFlow Object Detection API 超详细教程和踩坑过程(安装)
原文地址:https://www.cnblogs.com/yanshw/p/12598921.html