Use params.BASE_PATH closes #131

Author: mdangschat

asr/dataset_util/audio_sample_info.py

@@ -20,8 +20,9 @@
 
 # noinspection PyUnresolvedReferences
 from asr.load_sample import load_sample, WIN_STEP
+from asr.params import BASE_PATH
 
-DATASETS_PATH = '../datasets/speech_data'
+DATASETS_PATH = os.path.join(BASE_PATH, '../datasets/speech_data')
 
 
 def display_sample_info(file_path, label=''):
@@ -187,7 +188,7 @@ def display_sample_info(file_path, label=''):
 
 
 if __name__ == '__main__':
-    _test_txt_path = os.path.join('./data', 'train.txt')
+    _test_txt_path = os.path.join(BASE_PATH, 'data', 'train.txt')
 
     # Display specific sample info's.
     with open(_test_txt_path, 'r') as f:
@@ -196,7 +197,7 @@ def display_sample_info(file_path, label=''):
         # _line = _lines[37748]       # "The cat is on the roof"
         _line = _lines[375]
         _wav_path, txt = _line.split(' ', 1)
-        _wav_path = os.path.join('../datasets/speech_data', _wav_path)
+        _wav_path = os.path.join(DATASETS_PATH, _wav_path)
         _txt = txt.strip()
 
         display_sample_info(_wav_path, label=_txt)

asr/dataset_util/common_voice_loader.py

@@ -9,9 +9,8 @@
 from tqdm import tqdm
 from scipy.io import wavfile
 
-from asr.params import BASE_PATH
 from asr.util.storage import delete_file_if_exists
-from asr.params import MIN_EXAMPLE_LENGTH, MAX_EXAMPLE_LENGTH
+from asr.params import MIN_EXAMPLE_LENGTH, MAX_EXAMPLE_LENGTH, BASE_PATH
 
 
 # Path to the Mozilla Common Voice dataset.

asr/dataset_util/generate_txt.py

@@ -29,6 +29,7 @@
 import os
 import re
 
+from asr.params import BASE_PATH
 from asr.util import storage
 from asr.dataset_util.tatoeba_loader import tatoeba_loader
 from asr.dataset_util.timit_loader import timit_loader
@@ -38,7 +39,7 @@
 
 
 # Dataset base path.
-DATASET_PATH = '../datasets/speech_data'
+DATASET_PATH = os.path.join(BASE_PATH, '../datasets/speech_data')
 
 # Where to generate the .txt files, e.g. /home/user/../data/<target>.txt
 TXT_TARGET_PATH = './data/'

asr/dataset_util/libri_speech_loeader.py

@@ -4,8 +4,7 @@
 
 from scipy.io import wavfile
 
-from asr.params import BASE_PATH
-from asr.params import MIN_EXAMPLE_LENGTH, MAX_EXAMPLE_LENGTH
+from asr.params import MIN_EXAMPLE_LENGTH, MAX_EXAMPLE_LENGTH, BASE_PATH
 
 
 # Path to the LibriSpeech ASR dataset.

asr/dataset_util/plot_csv.py

@@ -14,5 +14,5 @@
 plt.grid(visible=True)
 plt.ylabel('WER')
 plt.xlabel('Step')
-plt.title('AIDS', visible=False)
+plt.title('TITLE', visible=False)
 plt.show()

asr/dataset_util/sd_estimator.py

@@ -11,9 +11,10 @@
 from tqdm import tqdm
 
 from asr.load_sample import load_sample
+from asr.params import BASE_PATH
 
 
-__DATASETS_PATH = '../datasets/speech_data'
+__DATASETS_PATH = os.path.join(BASE_PATH, '../datasets/speech_data')
 __FEATURE_TYPE = 'mel'
 
 
@@ -72,7 +73,7 @@ def __stat_calculator(line):
 
 if __name__ == '__main__':
     # Path to `train.txt` file.
-    _test_txt_path = os.path.join('./data', 'train.txt')
+    _test_txt_path = os.path.join(BASE_PATH, 'data', 'train.txt')
 
     # Display dataset stats.
     calculate_dataset_stats(_test_txt_path)

asr/dataset_util/sort_txt_by_seq_len.py

@@ -11,9 +11,10 @@
 from asr.util import storage
 from asr.load_sample import load_sample
 from asr.util.matplotlib_helper import pyplot_display
+from asr.params import BASE_PATH
 
 
-__DATASETS_PATH = '../datasets/speech_data'
+__DATASETS_PATH = os.path.join(BASE_PATH, '../datasets/speech_data')
 
 
 def _sort_txt_by_seq_len(txt_path, num_buckets=64, max_length=1700):

asr/dataset_util/tatoeba_loader.py

@@ -10,9 +10,8 @@
 from tqdm import tqdm
 from scipy.io import wavfile
 
-from asr.params import BASE_PATH
 from asr.util.storage import delete_file_if_exists
-from asr.params import MIN_EXAMPLE_LENGTH, MAX_EXAMPLE_LENGTH
+from asr.params import MIN_EXAMPLE_LENGTH, MAX_EXAMPLE_LENGTH, BASE_PATH
 
 
 # Path to the Taboeba dataset.

asr/dataset_util/tedlium_loader.py

@@ -9,9 +9,8 @@
 from tqdm import tqdm
 from scipy.io import wavfile
 
-from asr.params import BASE_PATH, FLAGS
+from asr.params import FLAGS, MIN_EXAMPLE_LENGTH, MAX_EXAMPLE_LENGTH, BASE_PATH
 from asr.util.storage import delete_file_if_exists
-from asr.params import MIN_EXAMPLE_LENGTH, MAX_EXAMPLE_LENGTH
 
 
 # Path to the Tedlium v2 dataset.

asr/dataset_util/timit_loader.py

@@ -4,8 +4,7 @@
 
 from scipy.io import wavfile
 
-from asr.params import BASE_PATH
-from asr.params import MIN_EXAMPLE_LENGTH, MAX_EXAMPLE_LENGTH
+from asr.params import MIN_EXAMPLE_LENGTH, MAX_EXAMPLE_LENGTH, BASE_PATH
 
 
 # Path to the TIMIT dataset.