Tensorflow Object Detection API
猪检测代码以及后续进行猪分类的程序都开源在github了。
主要在官方的demo code上做了如下修改:
- 扩展det出的box,以更好地包裹目标,crop时限定不超出图像边界[expand_ratio]
- 如检测出pig, animal可能都是对的,可以依据运行结果调整接受规则,抑制检测到的概率比较大的无关类别,提高鲁棒性[class_keep]
- 使用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