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.

__init__.py 2.4 kB

4 years ago
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. #! /usr/bin/python
  2. # -*- coding: utf-8 -*-
  3. """
  4. TensorLayer provides rich layer implementations trailed for
  5. various benchmarks and domain-specific problems. In addition, we also
  6. support transparent access to native TensorFlow parameters.
  7. For example, we provide not only layers for local response normalization, but also
  8. layers that allow user to apply ``tf.ops.lrn`` on ``network.outputs``.
  9. More functions can be found in `TensorFlow API <https://www.tensorflow.org/versions/master/api_docs/index.html>`__.
  10. """
  11. from tensorlayer.lazy_imports import LazyImport
  12. from .dataset_loaders.celebA_dataset import *
  13. from .dataset_loaders.cifar10_dataset import *
  14. from .dataset_loaders.cyclegan_dataset import *
  15. from .dataset_loaders.flickr_1M_dataset import *
  16. from .dataset_loaders.flickr_25k_dataset import *
  17. from .dataset_loaders.imdb_dataset import *
  18. from .dataset_loaders.matt_mahoney_dataset import *
  19. from .dataset_loaders.mnist_dataset import *
  20. from .dataset_loaders.mnist_fashion_dataset import *
  21. from .dataset_loaders.mpii_dataset import *
  22. from .dataset_loaders.nietzsche_dataset import *
  23. from .dataset_loaders.ptb_dataset import *
  24. from .dataset_loaders.voc_dataset import *
  25. from .dataset_loaders.wmt_en_fr_dataset import *
  26. from .utils import *
  27. __all__ = [
  28. # Dataset Loaders
  29. 'load_celebA_dataset',
  30. 'load_cifar10_dataset',
  31. 'load_cyclegan_dataset',
  32. 'load_fashion_mnist_dataset',
  33. 'load_flickr1M_dataset',
  34. 'load_flickr25k_dataset',
  35. 'load_imdb_dataset',
  36. 'load_matt_mahoney_text8_dataset',
  37. 'load_mnist_dataset',
  38. 'load_mpii_pose_dataset',
  39. 'load_nietzsche_dataset',
  40. 'load_ptb_dataset',
  41. 'load_voc_dataset',
  42. 'load_wmt_en_fr_dataset',
  43. # Util Functions
  44. 'assign_params',
  45. 'del_file',
  46. 'del_folder',
  47. 'download_file_from_google_drive',
  48. 'exists_or_mkdir',
  49. 'file_exists',
  50. 'folder_exists',
  51. 'load_and_assign_npz',
  52. 'load_and_assign_npz_dict',
  53. 'load_ckpt',
  54. 'load_cropped_svhn',
  55. 'load_file_list',
  56. 'load_folder_list',
  57. 'load_npy_to_any',
  58. 'load_npz',
  59. 'maybe_download_and_extract',
  60. 'natural_keys',
  61. 'npz_to_W_pdf',
  62. 'read_file',
  63. 'save_any_to_npy',
  64. 'save_ckpt',
  65. 'save_npz',
  66. 'save_npz_dict',
  67. 'load_and_assign_ckpt',
  68. 'ckpt_to_npz_dict'
  69. #'save_graph',
  70. #'load_graph',
  71. #'save_graph_and_params',
  72. #'load_graph_and_params',
  73. ]

TensorLayer3.0 是一款兼容多种深度学习框架为计算后端的深度学习库。计划兼容TensorFlow, Pytorch, MindSpore, Paddle.