SV-GAN/config.py at master · HPI-DeepLearning/SV-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
36
import os
from keras import backend as K

K.set_image_data_format('channels_last')

# general
data_dir = 'data/*'
test_dir = 'data/test/*'
label_dir= 'data/label.txt'
val_dir = 'data/val/*'
checkpoint_dir = 'checkpoints/'
validation_dir = 'data/validation/'
prediction_dir = 'data/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 = 110
sequence_length = 10
kernel_depth = 24

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