-
Notifications
You must be signed in to change notification settings - Fork 1
/
ProjectData.py
65 lines (52 loc) · 3.9 KB
/
ProjectData.py
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
class ProjectData:
def __init__(self):
# Files data
self.SOURCE_DIR = 'data/'
self.WAV_DIR = self.SOURCE_DIR + 'wav/'
self.WAV_TRAIN_DIR = self.WAV_DIR + 'wav_train/'
self.WAV_TEST_DIR = self.WAV_DIR + 'wav_test/'
self.TRANSCRIPTION_DIR = self.SOURCE_DIR + 'transcription/'
self.TRANSCRIPTION_TRAIN_DIR = self.TRANSCRIPTION_DIR + 'transcription_train/'
self.TRANSCRIPTION_TEST_DIR = self.TRANSCRIPTION_DIR + 'transcription_test/'
self.DATABASE_DIR = self.SOURCE_DIR
self.TRAIN_DATABASE_NAME = 'train_database.db'
self.VAL_DATABASE_NAME = 'validation_database.db'
self.TEST_DATABASE_NAME = 'test_database.db'
self.TRAIN_DATABASE_FILE = self.DATABASE_DIR + self.TRAIN_DATABASE_NAME
self.VAL_DATABASE_FILE = self.DATABASE_DIR + self.VAL_DATABASE_NAME
self.TEST_DATABASE_FILE = self.DATABASE_DIR + self.TEST_DATABASE_NAME
self.TFRECORD_TRAIN_DATABASE_FILE = self.DATABASE_DIR + 'train_database.tfrecords'
self.TFRECORD_VAL_DATABASE_FILE = self.DATABASE_DIR + 'val_database.tfrecords'
self.TFRECORD_TEST_DATABASE_FILE = self.DATABASE_DIR + 'test_database.tfrecords'
self.VOCAB_FILE = self.SOURCE_DIR + 'vocab_file.txt'
self.OUT_DIR = 'out/'
self.ZORZNET_CHECKPOINT_PATH = self.OUT_DIR + 'zorznet/' + 'checkpoint/model.ckpt'
self.ZORZNET_MODEL_PATH = self.OUT_DIR + 'zorznet/' + 'model/model'
self.ZORZNET_TENSORBOARD_PATH = self.OUT_DIR + 'zorznet/' + 'tensorboard/'
self.CTC_ATTENTION_CHECKPOINT_PATH = self.OUT_DIR + 'ctc_attention/' + 'checkpoint/model.ckpt'
self.CTC_ATTENTION_MODEL_PATH = self.OUT_DIR + 'ctc_attention/' + 'model/model'
self.CTC_ATTENTION_TENSORBOARD_PATH = self.OUT_DIR + 'ctc_attention/' + 'tensorboard/'
self.ZORZNET_ITERATED_CHECKPOINT_PATH = self.OUT_DIR + 'zorznet_iter/' + 'checkpoint/model.ckpt'
self.ZORZNET_ITERATED_MODEL_PATH = self.OUT_DIR + 'zorznet_iter/' + 'model/model'
self.ZORZNET_ITERATED_TENSORBOARD_PATH = self.OUT_DIR + 'zorznet_iter/' + 'tensorboard/'
self.ENC_DEC_CHECKPOINT_PATH = self.OUT_DIR + 'enc_dec/' + 'checkpoint/model.ckpt'
self.ENC_DEC_MODEL_PATH = self.OUT_DIR + 'enc_dec/' + 'model/model'
self.ENC_DEC_TENSORBOARD_PATH = self.OUT_DIR + 'enc_dec/' + 'tensorboard/'
self.REC_ENC_DEC_CHECKPOINT_PATH = self.OUT_DIR + 'rec_enc_dec/' + 'checkpoint/model.ckpt'
self.REC_ENC_DEC_MODEL_PATH = self.OUT_DIR + 'rec_enc_dec/' + 'model/model'
self.REC_ENC_DEC_TENSORBOARD_PATH = self.OUT_DIR + 'rec_enc_dec/' + 'tensorboard/'
self.ITERATED_CTC_CHECKPOINT_PATH = self.OUT_DIR + 'iterated_ctc/' + 'checkpoint/model.ckpt'
self.ITERATED_CTC_MODEL_PATH = self.OUT_DIR + 'iterated_ctc/' + 'model/model'
self.ITERATED_CTC_TENSORBOARD_PATH = self.OUT_DIR + 'iterated_ctc/' + 'tensorboard/'
self.ZORZNET_WORD_CTC_CHECKPOINT_PATH = self.OUT_DIR + 'zorznet_word_ctc/' + 'checkpoint/model.ckpt'
self.ZORZNET_WORD_CTC_MODEL_PATH = self.OUT_DIR + 'zorznet_word_ctc/' + 'model/model'
self.ZORZNET_WORD_CTC_TENSORBOARD_PATH = self.OUT_DIR + 'zorznet_word_ctc/' + 'tensorboard/'
self.LAS_NET_CHECKPOINT_PATH = self.OUT_DIR + 'las_net/' + 'checkpoint/model.ckpt'
self.LAS_NET_MODEL_PATH = self.OUT_DIR + 'las_net/' + 'model/model'
self.LAS_NET_TENSORBOARD_PATH = self.OUT_DIR + 'las_net/' + 'tensorboard/'
self.TRAIN_ENCODER_DATABASE_NAME = 'encoder_train_database.db'
self.VAL_ENCODER_DATABASE_NAME = 'encoder_validation_database.db'
self.TEST_ENCODER_DATABASE_NAME = 'encoder_test_database.db'
self.TRAIN_ENCODER_DATABASE_FILE = self.DATABASE_DIR + self.TRAIN_ENCODER_DATABASE_NAME
self.VAL_ENCODER_DATABASE_FILE = self.DATABASE_DIR + self.VAL_ENCODER_DATABASE_NAME
self.TEST_ENCODER_DATABASE_FILE = self.DATABASE_DIR + self.TEST_ENCODER_DATABASE_NAME