You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

README-CN.md 11 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228
  1. ![logo](assets/tf.net.logo.png)
  2. **Tensorflow.NET**是AI框架[TensorFlow](https://www.tensorflow.org/)在.NET平台上的实现,支持C#和F#,可以用来搭建深度学习模型并进行训练和推理,并内置了Numpy API,可以用来进行其它科学计算。
  3. Tensorflow.NET并非对于Python的简单封装,而是基于C API的pure C#实现,因此使用时无需额外的环境,可以很方便地用NuGet直接安装使用。并且dotnet团队提供的[ML.NET](https://github.com/dotnet/machinelearning)也依赖于Tensorflow.NET,支持调用Tensorflow.NET进行训练和推理,可以很方便地融入.NET生态。
  4. 与tensorflow相同,Tensorflow.NET也内置了Keras这一高级API,只要在安装Tensorflow.NET的同时安装Tensorflow.Keras就可以使用,Keras支持以模块化的方式调用模型,给模型的搭建提供了极大的便利。
  5. [![Join the chat at https://gitter.im/publiclab/publiclab](https://badges.gitter.im/Join%20Chat.svg)](https://gitter.im/sci-sharp/community)
  6. [![Tensorflow.NET](https://ci.appveyor.com/api/projects/status/wx4td43v2d3f2xj6?svg=true)](https://ci.appveyor.com/project/Haiping-Chen/tensorflow-net)
  7. [![NuGet](https://img.shields.io/nuget/dt/TensorFlow.NET.svg)](https://www.nuget.org/packages/TensorFlow.NET)
  8. [![Documentation Status](https://readthedocs.org/projects/tensorflownet/badge/?version=latest)](https://tensorflownet.readthedocs.io/en/latest/?badge=latest)
  9. [![Badge](https://img.shields.io/badge/link-996.icu-red.svg)](https://996.icu/#/en_US)
  10. [![Binder](https://mybinder.org/badge_logo.svg)](https://mybinder.org/v2/gh/javiercp/BinderTF.NET/master?urlpath=lab)
  11. 中文 | [English](https://github.com/SciSharp/TensorFlow.NET#readme)
  12. *当前主分支与Tensorflow2.10版本相对应,支持Eager Mode,同时也支持v1的静态图。*
  13. ![tensors_flowing](assets/tensors_flowing.gif)
  14. ## Why Tensorflow.NET?
  15. `SciSharp STACK`开源社区的目标是构建.NET平台下易用的科学计算库,而Tensorflow.NET就是其中最具代表性的仓库之一。在深度学习领域Python是主流,无论是初学者还是资深开发者,模型的搭建和训练都常常使用Python写就的AI框架,比如tensorflow。但在实际应用深度学习模型的时候,又可能希望用到.NET生态,亦或只是因为.NET是自己最熟悉的领域,这时候Tensorflow.NET就有显著的优点,因为它不仅可以和.NET生态很好地贴合,其API还使得开发者很容易将Python代码迁移过来。下面的对比就是很好的例子,Python代码和C#代码有着高度相似的API,这会使得迁移的时候无需做过多修改。
  16. ![python vs csharp](assets/syntax-comparision.png)
  17. 除了高度相似的API外,Tensorflow.NET与tensorflow也已经打通数据通道,tensorflow训练并保存的模型可以在Tensorflow.NET中直接读取并继续训练或推理,反之Tensorflow.NET保存的模型也可以在tensorflow中读取,这大大方便了模型的训练和部署。
  18. 与其它类似的库比如[TensorFlowSharp](https://www.nuget.org/packages/TensorFlowSharp/)相比,Tensorflow.NET的实现更加完全,提供了更多的高级API,使用起来更为方便,更新也更加迅速。
  19. ## 文档
  20. 基本介绍与简单用例:[Tensorflow.NET Documents](https://scisharp.github.io/tensorflow-net-docs)
  21. 详细文档:[The Definitive Guide to Tensorflow.NET](https://tensorflownet.readthedocs.io/en/latest/FrontCover.html)
  22. 例程:[TensorFlow.NET Examples](https://github.com/SciSharp/TensorFlow.NET-Examples)
  23. 运行例程常见问题:[Tensorflow.NET FAQ](tensorflowlib/README.md)
  24. ## 安装与使用
  25. 安装可以在NuGet包管理器中搜索包名安装,也可以用下面命令行的方式。
  26. 安装分为两个部分,第一部分是Tensorflow.NET的主体:
  27. ```sh
  28. ### 安装Tensorflow.NET
  29. PM> Install-Package TensorFlow.NET
  30. ### 安装Tensorflow.Keras
  31. PM> Install-Package TensorFlow.Keras
  32. ```
  33. 第二部分是计算支持部分,只需要根据自己的设备和系统选择下面之一即可:
  34. ```
  35. ### CPU版本,支持Windows、Linux和Mac
  36. PM> Install-Package SciSharp.TensorFlow.Redist
  37. ### Windows下的GPU版本(需要安装CUDA和cuDNN)
  38. PM> Install-Package SciSharp.TensorFlow.Redist-Windows-GPU
  39. ### Linux下的GPU版本(需要安装CUDA和cuDNN)
  40. PM> Install-Package SciSharp.TensorFlow.Redist-Linux-GPU
  41. ```
  42. 下面给出两个简单的例子,更多例子可以在[TensorFlow.NET Examples]中查看。
  43. ### 简单例子(使用Eager Mode进行线性回归)
  44. ```csharp
  45. using static Tensorflow.Binding;
  46. using static Tensorflow.KerasApi;
  47. using Tensorflow;
  48. using Tensorflow.NumPy;
  49. // Parameters
  50. var training_steps = 1000;
  51. var learning_rate = 0.01f;
  52. var display_step = 100;
  53. // Sample data
  54. var X = np.array(3.3f, 4.4f, 5.5f, 6.71f, 6.93f, 4.168f, 9.779f, 6.182f, 7.59f, 2.167f,
  55. 7.042f, 10.791f, 5.313f, 7.997f, 5.654f, 9.27f, 3.1f);
  56. var Y = np.array(1.7f, 2.76f, 2.09f, 3.19f, 1.694f, 1.573f, 3.366f, 2.596f, 2.53f, 1.221f,
  57. 2.827f, 3.465f, 1.65f, 2.904f, 2.42f, 2.94f, 1.3f);
  58. var n_samples = X.shape[0];
  59. // We can set a fixed init value in order to demo
  60. var W = tf.Variable(-0.06f, name: "weight");
  61. var b = tf.Variable(-0.73f, name: "bias");
  62. var optimizer = keras.optimizers.SGD(learning_rate);
  63. // Run training for the given number of steps.
  64. foreach (var step in range(1, training_steps + 1))
  65. {
  66. // Run the optimization to update W and b values.
  67. // Wrap computation inside a GradientTape for automatic differentiation.
  68. using var g = tf.GradientTape();
  69. // Linear regression (Wx + b).
  70. var pred = W * X + b;
  71. // Mean square error.
  72. var loss = tf.reduce_sum(tf.pow(pred - Y, 2)) / (2 * n_samples);
  73. // should stop recording
  74. // Compute gradients.
  75. var gradients = g.gradient(loss, (W, b));
  76. // Update W and b following gradients.
  77. optimizer.apply_gradients(zip(gradients, (W, b)));
  78. if (step % display_step == 0)
  79. {
  80. pred = W * X + b;
  81. loss = tf.reduce_sum(tf.pow(pred - Y, 2)) / (2 * n_samples);
  82. print($"step: {step}, loss: {loss.numpy()}, W: {W.numpy()}, b: {b.numpy()}");
  83. }
  84. }
  85. ```
  86. 这一用例也可以在[Jupyter Notebook Example](https://github.com/SciSharp/SciSharpCube)进行运行.
  87. ### 简单例子(使用Keras搭建Resnet)
  88. ```csharp
  89. using static Tensorflow.Binding;
  90. using static Tensorflow.KerasApi;
  91. using Tensorflow;
  92. using Tensorflow.NumPy;
  93. var layers = keras.layers;
  94. // input layer
  95. var inputs = keras.Input(shape: (32, 32, 3), name: "img");
  96. // convolutional layer
  97. var x = layers.Conv2D(32, 3, activation: "relu").Apply(inputs);
  98. x = layers.Conv2D(64, 3, activation: "relu").Apply(x);
  99. var block_1_output = layers.MaxPooling2D(3).Apply(x);
  100. x = layers.Conv2D(64, 3, activation: "relu", padding: "same").Apply(block_1_output);
  101. x = layers.Conv2D(64, 3, activation: "relu", padding: "same").Apply(x);
  102. var block_2_output = layers.Add().Apply(new Tensors(x, block_1_output));
  103. x = layers.Conv2D(64, 3, activation: "relu", padding: "same").Apply(block_2_output);
  104. x = layers.Conv2D(64, 3, activation: "relu", padding: "same").Apply(x);
  105. var block_3_output = layers.Add().Apply(new Tensors(x, block_2_output));
  106. x = layers.Conv2D(64, 3, activation: "relu").Apply(block_3_output);
  107. x = layers.GlobalAveragePooling2D().Apply(x);
  108. x = layers.Dense(256, activation: "relu").Apply(x);
  109. x = layers.Dropout(0.5f).Apply(x);
  110. // output layer
  111. var outputs = layers.Dense(10).Apply(x);
  112. // build keras model
  113. var model = keras.Model(inputs, outputs, name: "toy_resnet");
  114. model.summary();
  115. // compile keras model in tensorflow static graph
  116. model.compile(optimizer: keras.optimizers.RMSprop(1e-3f),
  117. loss: keras.losses.SparseCategoricalCrossentropy(from_logits: true),
  118. metrics: new[] { "acc" });
  119. // prepare dataset
  120. var ((x_train, y_train), (x_test, y_test)) = keras.datasets.cifar10.load_data();
  121. // normalize the input
  122. x_train = x_train / 255.0f;
  123. // training
  124. model.fit(x_train[new Slice(0, 2000)], y_train[new Slice(0, 2000)],
  125. batch_size: 64,
  126. epochs: 10,
  127. validation_split: 0.2f);
  128. // save the model
  129. model.save("./toy_resnet_model");
  130. ```
  131. 此外,Tensorflow.NET也支持用F#搭建上述模型进行训练和推理。
  132. ## Tensorflow.NET版本对应关系
  133. | TensorFlow.NET Versions | tensorflow 1.14, cuda 10.0 | tensorflow 1.15, cuda 10.0 | tensorflow 2.3, cuda 10.1 | tensorflow 2.4, cuda 11 | tensorflow 2.7, cuda 11 |tensorflow 2.10, cuda 11 |
  134. | -------------------------- | ------------- | -------------- | ------------- | ------------- | ------------ | ------------ |
  135. | tf.net 0.10x, tf.keras 0.10 | | | | | | x |
  136. | tf.net 0.7x, tf.keras 0.7 | | | | | x | |
  137. | tf.net 0.4x, tf.keras 0.5 | | | | x | | |
  138. | tf.net 0.3x, tf.keras 0.4 | | | x | | | |
  139. | tf.net 0.2x | | x | x | | | |
  140. | tf.net 0.15 | x | x | | | | |
  141. | tf.net 0.14 | x | | | | | |
  142. ```
  143. tf.net 0.4x -> tf native 2.4
  144. tf.net 0.6x -> tf native 2.6
  145. tf.net 0.7x -> tf native 2.7
  146. tf.net 0.10x -> tf native 2.10
  147. ...
  148. ```
  149. 如果使用过程中发现有缺失的版本,请告知我们,谢谢!
  150. 请注意Tensorflow.NET与Tensorflow.Keras版本存在一一对应关系,请安装与Tensorflow.NET对应的Tensorflow.Keras版本。
  151. ## 参与我们的开发:
  152. 我们欢迎任何人的任何形式的贡献!无论是文档中的错误纠正,新特性提议,还是BUG修复等等,都会使得Tensorflow.NET项目越来越好,Tensorflow.NET的全体开发者也会积极帮助解决您提出的问题。
  153. 下面任何一种形式都可以帮助Tensorflow.NET越来越好:
  154. * Star和分享Tensorflow.NET项目
  155. * 为Tensorflow.NET添加更多的用例
  156. * 在issue中告知我们Tensorflow.NET目前相比tensorflow缺少的API或者没有对齐的特性
  157. * 在issue中提出Tensorflow.NET存在的BUG或者可以改进的地方
  158. * 在待办事项清单中选择一个进行或者解决某个issue
  159. * 帮助我们完善文档,这也十分重要
  160. ## 支持我们
  161. 我们推出了[TensorFlow.NET实战](https://item.jd.com/13441549.html)这本书,包含了Tensorflow.NET主要开发者编写的讲解与实战例程,欢迎您的购买,希望这本书可以给您带来帮助。
  162. <p float="left">
  163. <img src="https://user-images.githubusercontent.com/1705364/198852429-91741881-c196-401e-8e9e-2f8656196613.png" width="250" />
  164. <img src="https://user-images.githubusercontent.com/1705364/198852521-2f842043-3ace-49d2-8533-039c6a043a3f.png" width="260" />
  165. <img src="https://user-images.githubusercontent.com/1705364/198852721-54cd9e7e-9210-4931-a86c-77584b25b8e1.png" width="260" />
  166. </p>
  167. ## 联系我们
  168. 可以在 [Twitter](https://twitter.com/ScisharpStack), [Facebook](https://www.facebook.com/scisharp.stack.9), [Medium](https://medium.com/scisharp), [LinkedIn](https://www.linkedin.com/company/scisharp-stack/)中关注我们,也可以在[Gitter](https://gitter.im/sci-sharp/community)中与项目开发者以及其它使用者进行沟通交流,也欢迎在仓库中提起issue。
  169. TensorFlow.NET is a part of [SciSharp STACK](https://scisharp.github.io/SciSharp/)
  170. <br>
  171. <a href="http://scisharpstack.org"><img src="https://github.com/SciSharp/SciSharp/blob/master/art/scisharp-stack.png" width="391" height="100" /></a>