forked from TensorLayer/tensorlayer3
78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
#! /usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
TensorLayer provides rich layer implementations trailed for
|
|
various benchmarks and domain-specific problems. In addition, we also
|
|
support transparent access to native TensorFlow parameters.
|
|
For example, we provide not only layers for local response normalization, but also
|
|
layers that allow user to apply ``tf.ops.lrn`` on ``network.outputs``.
|
|
More functions can be found in `TensorFlow API <https://www.tensorflow.org/versions/master/api_docs/index.html>`__.
|
|
"""
|
|
|
|
from tensorlayer.lazy_imports import LazyImport
|
|
|
|
from .dataset_loaders.celebA_dataset import *
|
|
from .dataset_loaders.cifar10_dataset import *
|
|
from .dataset_loaders.cyclegan_dataset import *
|
|
from .dataset_loaders.flickr_1M_dataset import *
|
|
from .dataset_loaders.flickr_25k_dataset import *
|
|
from .dataset_loaders.imdb_dataset import *
|
|
from .dataset_loaders.matt_mahoney_dataset import *
|
|
from .dataset_loaders.mnist_dataset import *
|
|
from .dataset_loaders.mnist_fashion_dataset import *
|
|
from .dataset_loaders.mpii_dataset import *
|
|
from .dataset_loaders.nietzsche_dataset import *
|
|
from .dataset_loaders.ptb_dataset import *
|
|
from .dataset_loaders.voc_dataset import *
|
|
from .dataset_loaders.wmt_en_fr_dataset import *
|
|
from .utils import *
|
|
|
|
__all__ = [
|
|
# Dataset Loaders
|
|
'load_celebA_dataset',
|
|
'load_cifar10_dataset',
|
|
'load_cyclegan_dataset',
|
|
'load_fashion_mnist_dataset',
|
|
'load_flickr1M_dataset',
|
|
'load_flickr25k_dataset',
|
|
'load_imdb_dataset',
|
|
'load_matt_mahoney_text8_dataset',
|
|
'load_mnist_dataset',
|
|
'load_mpii_pose_dataset',
|
|
'load_nietzsche_dataset',
|
|
'load_ptb_dataset',
|
|
'load_voc_dataset',
|
|
'load_wmt_en_fr_dataset',
|
|
|
|
# Util Functions
|
|
'assign_params',
|
|
'del_file',
|
|
'del_folder',
|
|
'download_file_from_google_drive',
|
|
'exists_or_mkdir',
|
|
'file_exists',
|
|
'folder_exists',
|
|
'load_and_assign_npz',
|
|
'load_and_assign_npz_dict',
|
|
'load_ckpt',
|
|
'load_cropped_svhn',
|
|
'load_file_list',
|
|
'load_folder_list',
|
|
'load_npy_to_any',
|
|
'load_npz',
|
|
'maybe_download_and_extract',
|
|
'natural_keys',
|
|
'npz_to_W_pdf',
|
|
'read_file',
|
|
'save_any_to_npy',
|
|
'save_ckpt',
|
|
'save_npz',
|
|
'save_npz_dict',
|
|
'load_and_assign_ckpt',
|
|
'ckpt_to_npz_dict'
|
|
#'save_graph',
|
|
#'load_graph',
|
|
#'save_graph_and_params',
|
|
#'load_graph_and_params',
|
|
]
|