File size: 1,452 Bytes
79cf5f5 |
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 |
import librosa
import tqdm
import os
import pathlib
import click
def length(src: str):
if os.path.isfile(src) and src.endswith('.wav'):
return librosa.get_duration(filename=src) / 3600.
elif os.path.isdir(src):
total = 0
for ch in [os.path.join(src, c) for c in os.listdir(src)]:
total += length(ch)
return total
return 0
# noinspection PyShadowingBuiltins
@click.command(help='Validate segment lengths')
@click.option('--dir', required=True, help='Path to the segments directory')
def validate_lengths(dir):
dir = pathlib.Path(dir)
assert dir.exists() and dir.is_dir(), 'The chosen path does not exist or is not a directory.'
reported = False
filelist = list(dir.glob('*.wav'))
total_length = 0.
for file in tqdm.tqdm(filelist):
wave_seconds = librosa.get_duration(filename=str(file))
if wave_seconds < 2.:
reported = True
print(f'Too short! \'{file}\' has a length of {round(wave_seconds, 1)} seconds!')
if wave_seconds > 20.:
reported = True
print(f'Too long! \'{file}\' has a length of {round(wave_seconds, 1)} seconds!')
total_length += wave_seconds / 3600.
print(f'Found {len(filelist)} segments with total length of {round(total_length, 2)} hours.')
if not reported:
print('All segments have proper length.')
if __name__ == '__main__':
validate_lengths()
|