utils.py 文件源码

def load_audio(path, target_sr=16000):
    data, orig_sr = sf.read(path)
    if orig_sr != target_sr:
        return librosa.resample(data, orig_sr, target_sr)
    return data

# def _update_progress(progress):
#     print("\rProgress: [{0:50s}] {1:.1f}%".format('#' * int(progress * 50),
#                                                   progress * 100), end="")
#
#
# def create_manifest(data_path, manifest_path, ordered=True):
#     file_paths = []
#     wav_files = [os.path.join(dirpath, f)
#                  for dirpath, dirnames, files in os.walk(data_path)
#                  for f in fnmatch.filter(files, '*.wav')]
#     size = len(wav_files)
#     counter = 0
#     for file_path in wav_files:
#         file_paths.append(file_path.strip())
#         counter += 1
#         _update_progress(counter / float(size))
#     print('\n')
#     if ordered:
#         _order_files(file_paths)
#     counter = 0
#     with io.FileIO(manifest_path, "w") as file:
#         for wav_path in file_paths:
#             transcript_path = wav_path.replace(
#                 '/wav/', '/txt/').replace('.wav', '.txt')
#             sample = os.path.abspath(wav_path) + ',' + \
#                 os.path.abspath(transcript_path) + '\n'
#             file.write(sample)
#             counter += 1
#             _update_progress(counter / float(size))
#     print('\n')
#
#
# def _order_files(file_paths):
#     print("Sorting files by length...")
#
#     def func(element):
#         output = subprocess.check_output(
#             ['soxi -D %s' % element.strip()],
#             shell=True
#         )
#         return float(output)
#
#     file_paths.sort(key=func)