对tensorflow models zoo的介绍(tensorflow modles zoo)

tensorflow包含两个模块:

1、Object Detectionhttps://github.com/tensorflow/models/tree/master/research/object_detection(物体检测)

2、TensorFlow Official Modelshttps://github.com/tensorflow/tensorflow(源码)

打开tensorflow models zoo github链接:https://github.com/tensorflow/models

        我们提供了一组预先培训过的检测模型,包括Coco数据集Kitti数据集开放图像数据集AVA V2.1数据集不自然物种检测数据集。如果您对那些数据集中已经存在的类别感兴趣,那么这些模型对于开箱即用的推断很有用。在新数据集的培训中,它们对于初始化模型也很有用。 

  • COCO 数据集(微软开源的数据集)
  • Kitti数据集(自动驾驶场景)
  • Open Images数据集(谷歌开源的数据集)
  • AVA v2.1数据集(人类动作识别数据集)
  • iNaturalist Species Detection数据集


熟悉TensorFlow的人都知道,tf在Github上的主页是:tensorflow

这个主页下又有两个比较重要的repo(看star数就知道了),

分别是TensorFlow的源代码repo:tensorflow/tensorflow,还有一个tensorflow/models

后者tensorflow/models是Google官方用TensorFlow做的各种各样的模型,相当于示例代码

比如用于图像分类的Slim,深度文字OCR,以及用于NLP任务的句法分析模型syntaxnet,Seq2Seq with Attention等等等等。这次公布的Object Detection API同样是放在了tensorflow/models里。

再来说下这次公布的代码的实现方式。首先,对于目标检测这个任务来说,前面必须有一个像样的ImageNet图像分类模型来充当所谓的特征提取(Feature Extraction)层,比如VGG16、ResNet等网络结构。TensorFlow官方实现这些网络结构的项目是TensorFlow Slim,而这次公布的Object Detection API正是基于Slim的。Slim这个库公布的时间较早,不仅收录了AlexNet、VGG16、VGG19、Inception、ResNet这些比较经典的耳熟能详的卷积网络模型,还有Google自己搞的Inception-Resnet,MobileNet等。

我们在TensorFlow Object Detection API的官方安装指南(地址:tensorflow/models)中,可以看到这样一句代码:

# From tensorflow/models/
export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim

很显然,这就是钦点用Slim作特征抽取了。

另外,以Faster RCNN为例,之前在github上,可以找到各种各样非官方的TensorFlow实现,但是这些实现使用的特征抽取层都不是Slim,而是五花八门的什么都有,另外一方面实现代码大量copy自原始的caffe的实现:rbgirshick/py-faster-rcnn,这次公布的代码里已经一点也找不到原始caffe实现的痕迹了。最后,原来非官方的Object Detection实现的质量参差不齐,去年我调过一个Faster RCNN,过程比较痛苦,在运行之前疯狂debug了三天才勉强跑了起来。这次Google官方公布的Object Detection API别的不说,代码质量肯定是过的去的,因此以后应该不会有人再造TensorFlow下Faster RCNN、R-FCN、SSD的轮子了。

说完了代码,再简单来说下公布的模型。主要公布了5个在COCO上训练的网络。网络结构分别是SSD+MobileNet、SSD+Inception、R-FCN+ResNet101、Faster RCNN+ResNet101、Faster RCNN+Inception_ResNet。后期应该还会有更多的模型加入进来。

检测模型链接:https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md

tensorflow models zoo组成部分:

TensorFlow Models

This repository contains a number of different models implemented in TensorFlow:

1、The official models are a collection of example models that use TensorFlow's high-level APIs. They are intended to be well-maintained, tested, and kept up to date with the latest stable TensorFlow API. They should also be reasonably optimized for fast performance while still being easy to read. We especially recommend newer TensorFlow users to start here.

1、官方模型是使用TensorFlow的高级API的示例模型的集合。它们旨在得到良好的维护、测试,并与最新的稳定TensorFlow API保持同步。它们还应该进行合理优化,以实现快速性能,同时仍然易于阅读。我们特别推荐新的TensorFlow用户从这里开始

2、The research models are a large collection of models implemented in TensorFlow by researchers. They are not officially supported or available in release branches; it is up to the individual researchers to maintain the models and/or provide support on issues and pull requests.

2、研究模型是研究人员在TensorFlow中实现的大量模型集合。它们在发布分支中没有官方支持或可用;由单个研究人员维护模型和/或在问题和请求上提供支持。

3、The samples folder contains code snippets and smaller models that demonstrate features of TensorFlow, including code presented in various blog posts.

3、samples文件夹包含代码片段和较小的模型,演示TensorFlow的特性,包括各种博客文章中显示的代码。

4、The tutorials folder is a collection of models described in the TensorFlow tutorials.

4、教程文件夹是TensorFlow教程中描述的模型集合。

猜你喜欢

转载自blog.csdn.net/qq_30460949/article/details/88924412
zoo