使用tensorflow object detection API 进行物体识别

背景

物体识别是目前神经网络的一个比较热门的应用领域。尤其是RCCN,FAST-RCNN, FASTER-RCNN出现之后。google把当前比较热门的网络和论文都实现成了API(当然是基于tensorflow),还发布了一些训练好的模型。具体可以看github上面的文档 链接。这篇博客是我github上面一个小项目的说明文档,同时也记录我优化这个程序的过程

项目效果

使用tensorflow object detection API 进行物体识别

使用tensorflow object detection API 进行物体识别

Github

项目代码链接

当前状态

硬件配置

CPU: i5-7600k
memory: 8G
GPU: GTX 1060 , 8G RAM

识别速度

目前我使用三个进程,一个用来读取视频,一个用来进行物体识别,一个用来播放识别后的视频。
当前可以做到25 - 30 帧左右。但是GPU的利用率在30%左右。所以下一步就是提高识别速度