Mirrors/Voice-Cloning-App
1
0
Fork 0
mirror of https://github.com/voice-cloning-app/Voice-Cloning-App.git synced 2026-08-29 10:09:19 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix create_dataset CLI

Browse Source
This commit is contained in:
SirBitesalot
2021-11-28 18:20:32 +01:00
parent ee0c49793f
commit e0c4559199
1 changed files with 16 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
dataset/create_dataset.py
View File

@@ -4,16 +4,16 @@ from os.path import dirname, abspath
import shutil
import sys
import os
from training import DEFAULT_ALPHABET
sys.path.append(dirname(dirname(abspath(__file__))))
from training import DEFAULT_ALPHABET
from training.utils import load_symbols
from dataset.audio_processing import convert_audio
from dataset.clip_generator import clip_generator, MIN_LENGTH, MAX_LENGTH
from dataset.analysis import save_dataset_info
from dataset.transcribe import Silero
AUDIO_FOLDER = "wavs"
UNLABELLED_FOLDER = "unlabelled"
METADATA_FILE = "metadata.csv"
@@ -96,15 +96,18 @@ if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Generate dataset")
parser.add_argument("-t", "--text_path", help="Path to text file", type=str, required=True)
parser.add_argument("-a", "--audio_path", help="Path to audio file", type=str, required=True)
parser.add_argument(
"-f", "--forced_alignment_path", help="Path to forced alignment JSON", type=str, default="align.json"
)
parser.add_argument("-o", "--output_path", help="Path to save snippets", type=str, default="wavs")
parser.add_argument("-u", "--unlabelled_path", help="Path to save unlabelled clips", type=str, default="unlabelled")
parser.add_argument(
"-l", "--label_path", help="Path to save snippet labelling text file", type=str, default="metadata.csv"
)
parser.add_argument("-i", "--info_path", help="Path to save info file", type=str, default="info.json")
parser.add_argument("-lang", "--language", help="The language to use", type=str, default="English")
parser.add_argument("-s", "--symbol_path", help="Path to symbol/alphabet file", type=str, default=None)
args = parser.parse_args()
create_dataset(**vars(args), transcription_model=Silero())
if args.symbol_path:
symbols = load_symbols(args.symbol_path)
else:
symbols = DEFAULT_ALPHABET
create_dataset(
text_path=args.text_path,
audio_path=args.audio_path,
transcription_model=Silero(args.language),
output_folder=args.output_path,
symbols=symbols,
)