1 # Copyright 2017 Google Inc. All Rights Reserved. 2 # 3 # Licensed under the Apache License, Version 2.0 (the "License"); 4 # you may not use this file except in compliance with the License. 5 # You may obtain a copy of the License at 6 # 7 # http://www.apache.org/licenses/LICENSE-2.0 8 # 9 # Unless required by applicable law or agreed to in writing, software 10 # distributed under the License is distributed on an "AS IS" BASIS, 11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 # See the License for the specific language governing permissions and 13 # limitations under the License. 14 # ============================================================================== 15 """TFGAN is a lightweight library for training and evaluating GANs. 16 17 In addition to providing the infrastructure for easily training and evaluating 18 GANS, this library contains modules for a TFGAN-backed Estimator, 19 evaluation metrics, features (such as virtual batch normalization), and losses. 20 Please see README.md for details and usage. 21 """ 22 23 from __future__ import absolute_import 24 from __future__ import division 25 from __future__ import print_function 26 27 # Collapse TFGAN into a tiered namespace. 28 from tensorflow.contrib.gan.python import estimator 29 from tensorflow.contrib.gan.python import eval # pylint:disable=redefined-builtin 30 from tensorflow.contrib.gan.python import features 31 from tensorflow.contrib.gan.python import losses 32 from tensorflow.contrib.gan.python import namedtuples 33 from tensorflow.contrib.gan.python import train 34 35 # pylint: disable=unused-import,wildcard-import 36 from tensorflow.contrib.gan.python.namedtuples import * 37 from tensorflow.contrib.gan.python.train import * 38 # pylint: enable=unused-import,wildcard-import 39 40 from tensorflow.python.util.all_util import remove_undocumented 41 42 _allowed_symbols = [ 43 'estimator', 44 'eval', 45 'features', 46 'losses', 47 ] 48 _allowed_symbols += train.__all__ 49 _allowed_symbols += namedtuples.__all__ 50 remove_undocumented(__name__, _allowed_symbols) 51