美文网首页
tensorflow学习笔记-slim模块

tensorflow学习笔记-slim模块

作者: 飞天小小猫 | 来源:发表于2018-10-25 14:54 被阅读21次

slim是TF推出的一个轻量级的high-level API of TensorFlow,用起来很方便,啊哈哈~

TF-Slim 是 tensorflow 中定义、训练和评估复杂模型的轻量级库。tf-slim中的组件可以轻易地和原生 tensorflow 框架以及例如 tf.contrib.learn 这样的框架进行整合。

Github:https://github.com/tensorflow/tensorflow/tree/master/tensorflow/contrib/slim
可以看到,slim代码是放在contrib下的,还没转正。不过可以先玩起来哈~

在TF的modles仓库 https://github.com/tensorflow/models/tree/master/research/slim下,我们可以看到slim的具体应用。值得高兴的事,slim中已经实现了很多经典的图像分类模型,并提供各个模型在ImageNet上的预训练结果,我们可以直接下载微调参数就可以应用到自己的数据上了,非常nice~
不过这个目录下的代码不是official的,不一定能适用最新版的TF,可能需要自己修改一下代码。

这里转载一下slim的api详细的介绍:
tensorflow中slim模块api介绍

相关文章

网友评论

      本文标题:tensorflow学习笔记-slim模块

      本文链接:https://www.haomeiwen.com/subject/tzzqtqtx.html