extern/Thorsten-Voice
1
0
Fork 1
mirror of https://github.com/thorstenMueller/Thorsten-Voice.git synced 2024-11-24 17:03:11 +01:00
Code Issues Packages Projects Releases Wiki Activity

Making ffmpeg conversion optional.

Browse Source
This commit is contained in:
Thorsten Mueller 2022-02-15 17:28:40 +01:00
parent c35507b1f7
commit 77ad01d4ff
1 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
helperScripts/MRS2LJSpeech.py
View File

@ -11,11 +11,11 @@
# v1.2 - Added choice for choosing which recording session should be exported as LJSpeech # v1.2 - Added choice for choosing which recording session should be exported as LJSpeech
# v1.3 - Added parameter mrs_dir to pass directory of Mimic-Recording-Studio # v1.3 - Added parameter mrs_dir to pass directory of Mimic-Recording-Studio
# v1.4 - Script won't crash when audio recorded has been deleted on disk # v1.4 - Script won't crash when audio recorded has been deleted on disk
# v1.5 - Added parameter "ffmpeg" to make converting with ffmpeg optional
from genericpath import exists from genericpath import exists
import glob import glob
import sqlite3 import sqlite3
import ffmpeg
import os import os
import argparse import argparse
import sys import sys
@ -81,6 +81,17 @@ def convert_audio():
# Remove Temp Folder # Remove Temp Folder
rmtree(output_dir_audio_temp) rmtree(output_dir_audio_temp)
def copy_audio():
global output_dir_audio
print('→ Using ffmpeg to convert recordings')
recordings = len([name for name in os.listdir(output_dir_audio_temp) if os.path.isfile(os.path.join(output_dir_audio_temp,name))])
print('→ Copy %s Audio Files to LJSpeech Dataset\n' % "{:,}".format(recordings))
for idx, wav in enumerate(glob.glob(os.path.join(output_dir_audio_temp, "*.wav"))):
shutil.copyfile(wav,os.path.join(output_dir_audio, os.path.basename(wav)))
def create_meta_data(mrs_dir): def create_meta_data(mrs_dir):
print('→ Creating META Data') print('→ Creating META Data')
@ -117,6 +128,7 @@ def create_meta_data(mrs_dir):
def main(): def main():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--mrs_dir', required=True) parser.add_argument('--mrs_dir', required=True)
parser.add_argument('--ffmpeg', required=False, default=False)
args = parser.parse_args() args = parser.parse_args()
if not os.path.isdir(os.path.join(args.mrs_dir,"backend")): if not os.path.isdir(os.path.join(args.mrs_dir,"backend")):
@ -126,7 +138,13 @@ def main():
create_folders() create_folders()
create_meta_data(args.mrs_dir) create_meta_data(args.mrs_dir)
convert_audio()
if(args.ffmpeg):
import ffmpeg
convert_audio()
else:
copy_audio()
print('\n\033[38;5;86;1m✔\033[0m COMPLETE【ツ】\n') print('\n\033[38;5;86;1m✔\033[0m COMPLETE【ツ】\n')