def main():
train_images, test_images, train_labels, test_labels = dataset.train_test_data(
'./PetImages_resize', batch_size)
train_size = len(train_labels)
test_size = len(test_labels)
train_dataset = dataset.create_dataset(train_images, train_labels)
train_dataset = train_dataset.cache().shuffle(
buffer_size=train_size).batch(batch_size).repeat(
num_epoch).make_one_shot_iterator().get_next()
test_dataset = dataset.create_dataset(test_images, test_labels)
test_dataset = test_dataset.cache().shuffle(
buffer_size=10).batch(test_size).make_one_shot_iterator().get_next()
with tf.Session() as sess:
model = DCGAN(sess,
train_dataset=train_dataset,
test_dataset=test_dataset,
train_size=train_size,
test_size=test_size,
batch_size=batch_size,
num_epoch=num_epoch)
model.build_model()
model.intialize_variables()
#model.create_image_from_generator()
model.train()
