TensorLayer 是基于 Google TensorFlow 开发的深度学习与增强学习库。它提供主流的深度学习与增强学习模块,可以非常容易地自定义模型以解决人工智能问题。
- [Apr] 一行代码下载及可视化MPII数据集
- [Apr] 发布models APIs提供预训练模型
- [Mar] 发布试验版本API以搭建量化网络
- [Jan] 出版《深度学习:一起玩转TensorLayer》
- [Dec] 官方支持分布式训练,见小例子.
- [Nov] 发布大量关于目标检测的数据增强APIs, 见tl.prepro.
- [Nov] 支持Convolutional LSTM, 见ConvLSTMLayer.
- [Nov] 支持Deformable Convolution, 见DeformableConv2dLayer.
- [Nov] 一行代码下载VOC数据集, 见load_voc_dataset.
- 恭喜TensorLayer团队获得ACM Multimedia年度最佳开源软件奖
- 新例子 Chatbot in 200 lines of code for Seq2Seq.
- 欢迎加入 Slack 讨论组.
- 发布 Sub-pixel Convolution 用于超分辨率复原.
- 你现在可以把 TF-Slim 和 Keras 导入到TensorLayer中了!
TensorLayer grow out from a need to combine the power of TensorFlow with the right building modules for deep neural networks. According to our years of research and practical experiences of tackling real-world machine learning problems, we come up with three design goals for TensorLayer:
- Simplicity: we make TensorLayer easy to work with by providing mass tutorials that can be deployed and run through in minutes. A TensorFlow user may find it easier to bootstrap with the simple, high-level APIs provided by TensorLayer, and then deep dive into their implementation details if need.
- Flexibility: developing an effective DL algorithm for a specific domain typically requires careful tunings from many aspects. Without the loss of simplicity, TensorLayer allows users to customize their modules by manipulating the native APIs of TensorFlow (e.g., training parameters, iteration control and tensor components).
- Performance: TensorLayer aims to provide zero-cost abstraction for TensorFlow. With its first-class support for TensorFlow, it can easily run on either heterogeneous platforms or multiple computation nodes without compromise in performance.
关于 TensorLayer 一个最常见的问题就是为什么我们需要开发这个库,与其他库如 Keras 和 Tflearn有什么区别。 TensorLayer 和这些库最大的区别在于灵活性和运行速度。深度学习用户会发现使用 Keras 和 Tflearn 能够非常快的上手(当然 TensorLayer 也提供与它们类似的简单 APIs),这些库提供高层抽象的API,对开发者隐藏了深度学习引擎的细节。这会让用户很难从底层中修改和优化,而这往往在特定领域时需要考虑的。尽管如此,灵活性不会导致效率的降低,TensorLayer 可以分布式和多样化部署以最优化运行速度。此外,TensorLayer 还能和很多库无缝使用,如 TF-Slim 等等。
简单来讲 TensorLayer 是一个适用于不同水平用户使用的库。对于初学者,TensorLayer 提供大量简单的API和大量的教程;对于中级用户,TensorLayer 的灵活性和透明性优势能大大体现出来(V1.2版本是很好的例子);对于高级用户,运行速度快和跨平台优势会体现出来。这样的好处是作为用户,我们不需要因为在不同的学习阶段,而去学不同的库了。
🌞🌞🌞 我们建议你在Github 上star和watch官方项目,这样当官方有更新时,你会立即知道。本文档为官方RTD文档的翻译版,更新速度会比英文原版慢,若你的英文还行,我们建议你直接阅读官方RTD文档
❤️❤️❤️ TensorLayer首批开发者包括中国人,我们承诺将一直支持中国社区
TensorLayer 运行需要 TensorFlow, numpy 和 matplotlib。 对于 GPU 加速,需要安装 CUDA 和 cuDNN。请在 这里 查看更多安装细节。
如果您已经安装过 TensorFlow,最简单的安装命令如下 (以英文Github为准):
[for stable version] pip install tensorlayer
[for master version] pip install git+https://github.com/zsdonghao/tensorlayer.git
第一个程序训练一个多层神经网络来对 MNIST 阿拉伯数字图片集进行分类。我们使用了简单了 scikit式函数,如 fit()
和 test()
。
import tensorflow as tf
import tensorlayer as tl
sess = tf.InteractiveSession()
# 准备数据
X_train, y_train, X_val, y_val, X_test, y_test = tl.files.load_mnist_dataset(shape=(-1,784))
# 定义 placeholder
x = tf.placeholder(tf.float32, shape=[None, 784], name='x')
y_ = tf.placeholder(tf.int64, shape=[None, ], name='y_')
# 定义模型
network = tl.layers.InputLayer(x, name='input_layer')
network = tl.layers.DropoutLayer(network, keep=0.8, name='drop1')
network = tl.layers.DenseLayer(network, n_units=800, act = tf.nn.relu, name='relu1')
network = tl.layers.DropoutLayer(network, keep=0.5, name='drop2')
network = tl.layers.DenseLayer(network, n_units=800, act = tf.nn.relu, name='relu2')
network = tl.layers.DropoutLayer(network, keep=0.5, name='drop3')
# 定义输出、损失函数和衡量指标
# tl.cost.cross_entropy 在内部使用 tf.nn.sparse_softmax_cross_entropy_with_logits() 实现 softmax
network = tl.layers.DenseLayer(network, n_units=10, act = tf.identity, name='output_layer')
y = network.outputs
cost = tl.cost.cross_entropy(y, y_,'cost')
correct_prediction = tf.equal(tf.argmax(y, 1), y_)
acc = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))
y_op = tf.argmax(tf.nn.softmax(y), 1)
# 定义 optimizer
train_params = network.all_params
train_op = tf.train.AdamOptimizer(learning_rate=0.0001
).minimize(cost, var_list=train_params)
# 初始化 session 中的所有参数
tl.layers.initialize_global_variables(sess)
# 列出模型信息
network.print_params()
network.print_layers()
# 训练模型, 但如果你想了解更多实现细节,或想成为机器学习领域的专家,
# 我们鼓励使用 tl.iterate.minibatches() 来训练模型
tl.utils.fit(sess, network, train_op, cost, X_train, y_train, x, y_,
acc=acc, batch_size=500, n_epoch=500, print_freq=5,
X_val=X_val, y_val=y_val, eval_train=False)
# 评估模型
tl.utils.test(sess, network, acc, X_test, y_test, x, y_, batch_size=None, cost=cost)
# 把模型保存成 .npz 文件
tl.files.save_npz(network.all_params , name='model.npz')
sess.close()
我们提供简单的 APIs 如 fit()
, test()
,这和 Scikit-learn, Keras 很相识,都是为了加快编程速度。不过,如果您想更好地控制训练过程,您可以在您的代码中使用 TensorFlow 原本的方法,如 sess.run
(tutorial_mnist.py
提供了一些例子)。
更多例子,请见 这里。
文档 [CN] [EN] [PDF] [Epub] [HTML] 不仅描述了如何使用 TensorLayer APIs,它还提供了大量的深度学习教程包括不同类型的神经网络、增强学习、自然语言处理等等。
我们还提供了 Google TensorFlow 网站例子的模块化实现,因此您还可以同时阅读 TensorFlow 的例子教程 [en] [cn] 。
TensorLayer is a major ongoing research project in Data Science Institute, Imperial College London. TensorLayer contributors are from Imperial College, Tsinghua University, Carnegie Mellon University, Google, Microsoft, Bloomberg and etc. The goal of the project is to develop a compositional language while complex learning systems can be build through composition of neural network modules. The whole development is now participated by numerous contributors here.
- 🇬🇧 If you are in London, we can discuss in person. Drop us an email to organize a meetup: tensorlayer@gmail.com.
- 🇨🇳 我们有官方的 中文文档。另外, 我们建立了多种交流渠道,如QQ群和微信群(申请入群时请star官方项目,并告知github用户名).
- 🇹🇭 เราขอเรียนเชิญนักพัฒนาคนไทยทุกคนที่สนใจจะเข้าร่วมทีมพัฒนา TensorLayer ติดต่อสอบถามรายละเอียดเพิ่มเติมได้ที่ tensorlayer@gmail.com.
TensorLayer is releazed under the Apache 2.0 license.