安装Google Object Dectection API
官网:https://github.com/tensorflow/models/tree/master/research/object_detection
- 安装python3.5
- 使用virtualenv创建基于python3.5的虚拟环境
- 安装tensorflow
- 安装各种依赖
此处完全按照官网操作即可
sudo apt-get install protobuf-compiler python-pil python-lxml python-tk
sudo pip install Cython
sudo pip install jupyter
sudo pip install matplotlib
sudo pip install Cython
sudo pip install pillow
sudo pip install lxml
sudo pip install jupyter
sudo pip install matplotlib
下载model
访问 https://github.com/tensorflow/models 下载到本地(我在阿里的机器上用git没操作成功,直接在浏览器里下载)
解压上传到服务器coco API
下载,上传到服务器
编译
把以下文件夹拷贝到model/research下编译protobuf
到research目录下执行以下- 编辑PYTHONPATH
- 测试
以上主要参考https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/installation.md 即可
识别测试
运行例子
- 在服务器上,进入object dectection目录启动jupyter-book
- 在本机打开notebook
把上图的红框里的地址放到地址栏里(要把0.0.0.0换成你机器的IP)
- 找到object_dectection_tutorial.ipynb
- 打开
- 运行:Cell、Run All
- 等十秒左右,最下面出现对示例图片的识别
识别自己的图片
看刚刚运行的代码,找到test_images,这就是获取图片的位置
在服务器上找到test_images文件夹,会找到两张自带图片,把自己的图片拷贝进去,最好用相同的名字替代
-识别了几张自己的,效果如下
可见缺省模型对骷髅的识别还不行:)