新智元导读】谷歌今天宣布开源 TensorFlow 高级软件包 TF-Slim,能使用户快速准确地定义复杂模型,尤其是图像分类任务。这不由让人想起 Facebook 上周开源“从像素级别理解图像”的计算机视觉系统。不管怎么说,在计算机视觉方面,强大的工具又多了。下文是官方博文翻译。在新智元后台回复“0831”下载论文。
今年早些时候,我们发布了图像分类模型 Inception V3 在 TensorFlow 上的运行案例。代码能够让用户使用同步梯度下降用 ImageNet 分类数据库训练模型。Inception V3 模型的基础是一个叫做 TF-Slim 的 TensorFlow 库,用户可以使用这个软件包定义、训练、评估 TensorFlow 模型。TF-Slim 库提供的常用抽象能使用户快速准确地定义模型,同时确保模型架构透明,超参数明确。
自发布以来,TF-Slim 已经得到长足发展,无论是网络层、代价函数,还是评估标准,都增加了很多类型,训练和评估模型也有了很多便利的常规操作手段。这些手段使你在并行读取数据或者在多台机器上部署模型等大规模运行时,不必为细节操心。此外,我们还制作了 TF-Slim 图像模型库,为很多广泛使用的图像分类模型提供了定义以及训练脚本,这些都是使用标准的数据库写就的。TF-Slim 及其组成部分都已经在谷歌内部得到广泛的使用,很多升级也都整合进了 tf.contrib.slim.
今天,我们将 TF-Slim 的最新版本与 TF 社区共享,亮点包括:
-
很多新的层(比如 Atrous 卷积层和 Deconvolution),使神经网络架构更丰富;
-
支持更多的代价函数和评估指标(如 mAP,IoU)
-
部署运行库,让在一台或多台机器上进行同步或异步训练更容易
-
代码,用于定义和训练广泛使用的图像分类模型,比如 Inception、VGG、AlexNet、ResNet
-
训练好的模型,这些模型使用 ImageNet 分类数据库训练,但也能用于其他计算机视觉任务
-
ImageNet、CIFAR10 和 MNIST 这些容易使用的标准图像数据库
使用 TF-Slim 的 GitHbu 代码:
README:https://github.com/tensorflow/models/blob/master/slim/README.md
使用说明:https://github.com/tensorflow/tensorflow/tree/master/tensorflow/contrib/slim