Update
This commit is contained in:
litagin02
@@ -13,9 +13,16 @@ with open(os.path.join("configs", "paths.yml"), "r", encoding="utf-8") as f:
|
||||
# assets_root = path_config["assets_root"]
|
||||
|
||||
|
||||
def do_slice(model_name: str, min_sec: float, max_sec: float, input_dir="inputs"):
|
||||
def do_slice(
|
||||
model_name: str,
|
||||
min_sec: float,
|
||||
max_sec: float,
|
||||
min_silence_dur_ms: int,
|
||||
input_dir: str,
|
||||
):
|
||||
if model_name == "":
|
||||
return "Error: モデル名を入力してください。"
|
||||
logger.info("Start slicing...")
|
||||
input_dir = "inputs"
|
||||
output_dir = os.path.join(dataset_root, model_name, "raw")
|
||||
cmd = [
|
||||
"slice.py",
|
||||
@@ -27,20 +34,28 @@ def do_slice(model_name: str, min_sec: float, max_sec: float, input_dir="inputs"
|
||||
str(min_sec),
|
||||
"--max_sec",
|
||||
str(max_sec),
|
||||
"--min_silence_dur_ms",
|
||||
str(min_silence_dur_ms),
|
||||
]
|
||||
# onnxの警告が出るので無視する
|
||||
success, message = run_script_with_log(cmd, ignore_warning=True)
|
||||
if not success:
|
||||
return f"Error: {message}"
|
||||
return "音声のスライスが完了しました。"
|
||||
|
||||
|
||||
def do_transcribe(model_name, whisper_model, compute_type, language, initial_prompt):
|
||||
def do_transcribe(
|
||||
model_name, whisper_model, compute_type, language, initial_prompt, input_dir, device
|
||||
):
|
||||
if model_name == "":
|
||||
return "Error: モデル名を入力してください。"
|
||||
if initial_prompt == "":
|
||||
initial_prompt = "こんにちは。元気、ですかー？私は……ちゃんと元気だよ！"
|
||||
initial_prompt = "こんにちは。元気、ですかー？私は……ふふっ、ちゃんと元気だよ！"
|
||||
logger.debug(f"initial_prompt: {initial_prompt}")
|
||||
input_dir = os.path.join(dataset_root, model_name, "raw")
|
||||
if input_dir == "":
|
||||
input_dir = os.path.join(dataset_root, model_name, "raw")
|
||||
output_file = os.path.join(dataset_root, model_name, "esd.list")
|
||||
result = run_script_with_log(
|
||||
success, message = run_script_with_log(
|
||||
[
|
||||
"transcribe.py",
|
||||
"--input_dir",
|
||||
@@ -53,29 +68,44 @@ def do_transcribe(model_name, whisper_model, compute_type, language, initial_pro
|
||||
whisper_model,
|
||||
"--compute_type",
|
||||
compute_type,
|
||||
"--device",
|
||||
device,
|
||||
"--language",
|
||||
language,
|
||||
"--initial_prompt",
|
||||
f'"{initial_prompt}"',
|
||||
]
|
||||
)
|
||||
if not success:
|
||||
return f"Error: {message}"
|
||||
return "音声の文字起こしが完了しました。"
|
||||
|
||||
|
||||
initial_md = """
|
||||
# 簡易学習用データセット作成ツール
|
||||
|
||||
Style-Bert-VITS2の学習用データセットを作成するためのツールです。与えられた音声からちょうどいい長さの発話区間を切り取りスライスし、それぞれの音声に対して文字起こしを行います。
|
||||
Style-Bert-VITS2の学習用データセットを作成するためのツールです。以下の2つからなります。
|
||||
|
||||
- 与えられた音声からちょうどいい長さの発話区間を切り取りスライス
|
||||
- 音声に対して文字起こし
|
||||
|
||||
このうち両方を使ってもよいし、スライスする必要がない場合は後者のみを使ってもよいです。
|
||||
|
||||
## 必要なもの
|
||||
|
||||
学習したい音声が入ったwavファイルいくつか。
|
||||
合計時間がある程度はあったほうがいいかも、10分とかでも大丈夫だったとの報告あり。単一ファイルでも良いし複数ファイルでもよい。
|
||||
|
||||
## 使い方
|
||||
1. `inputs`フォルダ直下にwavファイルをすべて入れる
|
||||
2. `モデル名`を入力して、`音声のスライス`ボタンを押す
|
||||
3. 完了したら、`音声の文字起こし`ボタンを押す
|
||||
4. 出来上がった音声ファイルたちは`Data/{モデル名}/raw`に、書き起こしファイルは`Data/{モデル名}/esd.list`に保存されます。
|
||||
## スライス使い方
|
||||
1. `inputs`フォルダにwavファイルをすべて入れる
|
||||
2. `モデル名`を入力して、設定を必要なら調整して`音声のスライス`ボタンを押す
|
||||
3. 出来上がった音声ファイルたちは`Data/{モデル名}/raw`に保存される
|
||||
|
||||
## 書き起こし使い方
|
||||
|
||||
1. 書き起こしたい音声ファイルのあるフォルダを指定（デフォルトは`Data/{モデル名}/raw`なのでスライス後に行う場合は省略してよい）
|
||||
2. 設定を必要なら調整してボタンを押す
|
||||
3. 書き起こしファイルは`Data/{モデル名}/esd.list`に保存される
|
||||
|
||||
## 注意
|
||||
|
||||
@@ -90,16 +120,31 @@ with gr.Blocks(theme="NoCrypt/miku") as app:
|
||||
with gr.Accordion("音声のスライス"):
|
||||
with gr.Row():
|
||||
with gr.Column():
|
||||
input_dir = gr.Textbox(
|
||||
label="入力フォルダ名（デフォルトはinputs）",
|
||||
placeholder="inputs",
|
||||
info="inputsフォルダにwavファイルを入れておいてください",
|
||||
)
|
||||
min_sec = gr.Slider(
|
||||
minimum=0, maximum=10, value=2, step=0.5, label="この秒数未満は切り捨てる"
|
||||
)
|
||||
max_sec = gr.Slider(
|
||||
minimum=0, maximum=15, value=12, step=0.5, label="この秒数以上は切り捨てる"
|
||||
)
|
||||
min_silence_dur_ms = gr.Slider(
|
||||
minimum=0,
|
||||
maximum=2000,
|
||||
value=700,
|
||||
step=100,
|
||||
label="無音とみなして区切る最小の無音の長さ（ms）",
|
||||
)
|
||||
slice_button = gr.Button("スライスを実行")
|
||||
result1 = gr.Textbox(label="結果")
|
||||
with gr.Row():
|
||||
with gr.Column():
|
||||
raw_dir = gr.Textbox(
|
||||
label="書き起こしたい音声ファイルが入っているフォルダ（スライスした場合など、`Data/{モデル名}/raw`の場合は省略可",
|
||||
)
|
||||
whisper_model = gr.Dropdown(
|
||||
["tiny", "base", "small", "medium", "large", "large-v2", "large-v3"],
|
||||
label="Whisperモデル",
|
||||
@@ -119,20 +164,31 @@ with gr.Blocks(theme="NoCrypt/miku") as app:
|
||||
label="計算精度",
|
||||
value="bfloat16",
|
||||
)
|
||||
device = gr.Radio(["cuda", "cpu"], label="デバイス", value="cuda")
|
||||
language = gr.Dropdown(["ja", "en", "zh"], value="ja", label="言語")
|
||||
initial_prompt = gr.Textbox(
|
||||
label="初期プロンプト（日本語の場合は省略可）", placeholder="こんにちは。元気、ですかー？私は……ちゃんと元気だよ！"
|
||||
label="初期プロンプト",
|
||||
placeholder="こんにちは。元気、ですかー？ふふっ、私は……ちゃんと元気だよ！",
|
||||
info="このように書き起こしてほしいという例文、日本語なら省略可、英語等なら書いてください",
|
||||
)
|
||||
transcribe_button = gr.Button("音声の文字起こし")
|
||||
result2 = gr.Textbox(label="結果")
|
||||
slice_button.click(
|
||||
do_slice,
|
||||
inputs=[model_name, min_sec, max_sec],
|
||||
inputs=[model_name, min_sec, max_sec, min_silence_dur_ms, input_dir],
|
||||
outputs=[result1],
|
||||
)
|
||||
transcribe_button.click(
|
||||
do_transcribe,
|
||||
inputs=[model_name, whisper_model, compute_type, language, initial_prompt],
|
||||
inputs=[
|
||||
model_name,
|
||||
whisper_model,
|
||||
compute_type,
|
||||
language,
|
||||
initial_prompt,
|
||||
raw_dir,
|
||||
device,
|
||||
],
|
||||
outputs=[result2],
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user