# Path: celebA_dcgan/model_def.py
# class Hparams(object):
#     def __init__(self):
# def generator(hparams, z, train, reuse):
# def discriminator(hparams, x, train, reuse):
# def gen_restore_vars():
# def discrim_restore_vars():


import os
import sys
import tensorflow as tf
from celebA_dcgan import model_def as celebA_dcgan_model_def
# pylint: disable = C0103, C0111, C0301, R0914

"""Model definitions for celebA

This file is partially based on
https://github.com/carpedm20/DCGAN-tensorflow/blob/master/main.py
https://github.com/carpedm20/DCGAN-tensorflow/blob/master/model.py

They come with the following license: https://github.com/carpedm20/DCGAN-tensorflow/blob/master/LICENSE
"""


sys.path.append(os.path.join(os.path.dirname(__file__), '..'))


def dcgan_discrim(x_hat_batch, hparams):

    assert hparams.batch_size in [1, 64], 'batch size should be either 64 or 1'
    x_hat_image = tf.reshape(x_hat_batch, [-1, 64, 64, 3])
    all_zeros = tf.zeros([64, 64, 64, 3])
    discrim_input = all_zeros + x_hat_image


# Path: celebA_dcgan/model_def.py
# class Hparams(object):
#     def __init__(self):
# def generator(hparams, z, train, reuse):
# def discriminator(hparams, x, train, reuse):
# def gen_restore_vars():
# def discrim_restore_vars():


# Path: src/celebA_model_def.py
import os
import sys
import tensorflow as tf
from celebA_dcgan import model_def as celebA_dcgan_model_def

# pylint: disable = C0103, C0111, C0301, R0914

"""Model definitions for celebA

This file is partially based on
https://github.com/carpedm20/DCGAN-tensorflow/blob/master/main.py
https://github.com/carpedm20/DCGAN-tensorflow/blob/master/model.py

They come with the following license: https://github.com/carpedm20/DCGAN-tensorflow/blob/master/LICENSE
"""


sys.path.append(os.path.join(os.path.dirname(__file__), '..'))


def dcgan_discrim(x_hat_batch, hparams):

    assert hparams.batch_size in [1, 64], 'batch size should be either 64 or 1'
    x_hat_image = tf.reshape(x_hat_batch, [-1, 64, 64, 3])
    all_zeros = tf.zeros([64, 64, 64, 3])
    discrim_input = all_zeros + x_hat_image


    model_hparams = celebA_dcgan_model_def.Hparams()