def get_audio_files():
audio_files = list(get_audio_files_from_dir(ARGS.data_dir))
log.info("Found {} audio files for training.".format(len(audio_files)))
if len(audio_files) == 0:
log.close()
exit(1)
return audio_files
def get_audio_files():
audio_files = None
if input_data.can_load_from_csv():
log.info("Found csv files in {}".format(ARGS.data_dir))
else:
log.debug("Searching for audio files in {}".format(ARGS.data_dir))
if ARGS.filter_broken_audio:
data_dir_ = pathlib.Path(ARGS.data_dir)
audio_files = get_audio_files_from_dir(ARGS.data_dir)
log.debug("Moving possibly broken audio files to .bkp:")
broken_files = get_broken_audio_files(audio_files, ARGS.data_dir)
for f in broken_files:
log.debug(f)
bkp_dir = data_dir_.joinpath(f).parent.joinpath(".bkp")
bkp_dir.mkdir(exist_ok=True)
f = pathlib.Path(f)
data_dir_.joinpath(f).rename(bkp_dir.joinpath(f.name))
audio_files = list(get_audio_files_from_dir(ARGS.data_dir))
log.info("Found {} audio files for training.".format(len(audio_files)))
if len(audio_files) == 0:
log.close()
exit(1)
return audio_files