object_detection_with_tensorflow

2017-10-23 本文已影响75人磐创AI_聊天机器人

使用google 物体识别的api来识别电影里面的物体。目前可以做到每秒30帧。scotthuang1989/object_detection_with_tensorflow

分享自 TensorFlow QQ 群的彼岸：314148034

Target

detect people in a video or from camera.
can choose what detection method we are using.
Project Structure

From google

object label and detecion api
data
object_detection
My module

myutil: help function
object_detection_tf.py: main module for detecting object
Usage

run following command
python object_detection_tf.py -v /home/scott/Videos/S11E03.mp4

you need replace the video file with your choice
Current Status

on my nvidia 1060 (6GB RAM ), the speed is ~25 frame per second, but gpu utilization is below 30%, I will use multiprocessing to speed up.