Recurrent-GAN/config.py at master · HPI-DeepLearning/Recurrent-GAN · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import os
from keras import backend as K

K.set_image_data_format('channels_last')

# general
data_dir = 'data/*'
test_dir = 'test/*'
val_dir = 'val/*'
checkpoint_dir = 'checkpoints/'
validation_dir = 'validation/'
prediction_dir = 'prediction/'

directory = os.path.dirname(checkpoint_dir)
if not os.path.exists(directory):
    os.makedirs(directory)

directory = os.path.dirname(validation_dir)
if not os.path.exists(directory):
    os.makedirs(directory)

directory = os.path.dirname(prediction_dir)
if not os.path.exists(directory):
    os.makedirs(directory)

# bdsscgan
input = 4
output = 4
size = 240
epochs = 300
sequence_length = 10
kernel_depth = 24

checkpoint_gen_name = checkpoint_dir + 'gen.hdf5'
checkpoint_disc_name = checkpoint_dir + 'disc.hdf5'