tensorflow object detection 模型测试

模型替换

# What model to download. MODEL_NAME = 'ssd_mobilenet_v1_coco_11_06_2017'ssd_inception_v2_coco_11_06_2017 MODEL_FILE = MODEL_NAME + '.tar.gz' DOWNLOAD_BASE = 'http://download.tensorflow.org/models/object_detection/' http://download.tensorflow.org/models/object_detection/ssd_inception_v2_coco_11_06_2017.tar.gz

 只需要修改红色部分

 

使用ssd_mobilenet_v1_coco_11_06_2017模型，效果不是很好。一些较小的目标无法识别出来。

 

同样使用mobilenet模型，dog识别成了cat，结果很感人。

 

ssd_inception_v2_coco_11_06_2017模型，结果就好很多。

 

ssd_inception_v2_coco_11_06_2017模型，效果要好一些。