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

Added --mrs_dir param for more flexibility

Browse Source
This commit is contained in:
Thorsten Mueller 2021-07-07 22:00:47 +02:00
parent 2ea44ede87
commit 4f875ac591
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
helperScripts/MRS2LJSpeech.py
View File

@ -9,18 +9,20 @@
# - See more details here: https://gist.github.com/manifestinteractive/6fd9be62d0ede934d4e1171e5e751aba # - See more details here: https://gist.github.com/manifestinteractive/6fd9be62d0ede934d4e1171e5e751aba
# - Thanks Peter, it's a great contribution :-) # - Thanks Peter, it's a great contribution :-)
# 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
import glob import glob
import sqlite3 import sqlite3
import ffmpeg import ffmpeg
import os import os
import argparse
import sys
from shutil import copyfile from shutil import copyfile
from shutil import rmtree from shutil import rmtree
# Setup Directory Data # Setup Directory Data
cwd = os.path.dirname(os.path.abspath(__file__)) cwd = os.path.dirname(os.path.abspath(__file__))
mrs_dir = os.path.join(cwd, os.pardir, "mimic-recording-studio")
output_dir = os.path.join(cwd, "dataset") output_dir = os.path.join(cwd, "dataset")
output_dir_audio = "" output_dir_audio = ""
output_dir_audio_temp="" output_dir_audio_temp=""
@ -77,7 +79,7 @@ def convert_audio():
# Remove Temp Folder # Remove Temp Folder
rmtree(output_dir_audio_temp) rmtree(output_dir_audio_temp)
def create_meta_data(): def create_meta_data(mrs_dir):
print('→ Creating META Data') print('→ Creating META Data')
conn = sqlite3.connect(os.path.join(mrs_dir, "backend", "db", "mimicstudio.db")) conn = sqlite3.connect(os.path.join(mrs_dir, "backend", "db", "mimicstudio.db"))
@ -107,10 +109,17 @@ def create_meta_data():
conn.close() conn.close()
def main(): def main():
parser = argparse.ArgumentParser()
parser.add_argument('--mrs_dir', required=True)
args = parser.parse_args()
if not os.path.isdir(os.path.join(args.mrs_dir,"backend")):
sys.exit("Passed directory is no valid Mimic-Recording-Studio main directory!")
print('\n\033[48;5;22m MRS to LJ Speech Processor \033[0m\n') print('\n\033[48;5;22m MRS to LJ Speech Processor \033[0m\n')
create_folders() create_folders()
create_meta_data() create_meta_data(args.mrs_dir)
convert_audio() convert_audio()
print('\n\033[38;5;86;1m✔\033[0m COMPLETE【ツ】\n') print('\n\033[38;5;86;1m✔\033[0m COMPLETE【ツ】\n')