使用tensorflow object detection API 进行物体识别
背景
物体识别是目前神经网络的一个比较热门的应用领域。尤其是RCCN,FAST-RCNN, FASTER-RCNN出现之后。google把当前比较热门的网络和论文都实现成了API(当然是基于tensorflow),还发布了一些训练好的模型。具体可以看github上面的文档 链接。这篇博客是我github上面一个小项目的说明文档,同时也记录我优化这个程序的过程
项目效果
Github
当前状态
硬件配置
CPU: i5-7600k
memory: 8G
GPU: GTX 1060 , 8G RAM
识别速度
目前我使用三个进程,一个用来读取视频,一个用来进行物体识别,一个用来播放识别后的视频。
当前可以做到25 - 30 帧左右。但是GPU的利用率在30%左右。所以下一步就是提高识别速度