使用Tensorflow物体识别API抠出视频中的猪

Tensorflow Object Detection API

猪检测代码以及后续进行猪分类的程序都开源在github了。

主要在官方的demo code上做了如下修改：

1. 扩展det出的box，以更好地包裹目标，crop时限定不超出图像边界[expand_ratio]
2. 如检测出pig, animal可能都是对的，可以依据运行结果调整接受规则，抑制检测到的概率比较大的无关类别，提高鲁棒性[class_keep]
3. 使用mini batch的方式，以充分利用GPU提高程序运行效率。

下面重点看一下与obj det API有关的核心代码：

# Load a (frozen) Tensorflow model into memory
'''
tf.GraphDef():
The GraphDef class is an object created by the ProtoBuf. 
详见https://www.tensorflow.org/extend/tool developers/
graph_def: 
A GraphDef proto containing operations to be imported into the de