专栏名称: 中国人工智能学会
关注中国人工智能学会官方公共账号,收取学会的科普信息、新闻动态、活动预告及人工智能领域科技前沿信息。
目录
相关文章推荐
51好读  ›  专栏  ›  中国人工智能学会

手把手 | 亲测好用!Google发布了一个新的Tensorflow物体识别API

中国人工智能学会  · 公众号  · AI  · 2017-08-31 17:03

正文

请到「今天看啥」查看全文



COCO数据集的一些物体种类


这个API提供了5种不同的模型,使用者可以通过设置不同检测边界范围来平衡运行速度和准确率。



上图中的mAP(平均精度)是检测边界框的准确率和回召率的乘积。这是一个很好的混合测度,在评价模型对目标物体的敏锐度和它是否能很好的避免虚假目标中非常好用。mAP值越高,模型的准确度越高,但运行速度会相应下降。

(想要了解更多跟模型有关的知识https://github.com/tensorflow/models/blob/477ed41e7e4e8a8443bc633846eb01e2182dc68a/object_detection/g3doc/detection_model_zoo.md)


实测时间


我决定使用最轻量级的模型(ssd_mobilenet)。主要步骤如下:


1. 下载一个打包模型(.pb-protobuf)并把它载入缓存

2. 使用内置的辅助代码来载入标签,类别,可视化工具等等。

3. 建立一个新的会话,在图片上运行模型。


总体来说步骤非常简单。而且这个API文档还提供了一些能运行这些主要步骤的Jupyter文档——

https://github.com/tensorflow/models/blob/master/object_detection/object_detection_tutorial.ipynb


这个模型在实例图像上表现得相当出色(如下图):







请到「今天看啥」查看全文