pydomino
は日本語音声に対して音素ラベルをアラインメントするためのツールです。GPUは不要です。
ライブラリとして Python から使うこともコマンドラインツールとしてコンソールから使うこともできます。
ドキュメントは こちら からご覧いただけます。
技術の解説記事は こちら からご覧いただけます
- CMake
- Python >= 3.10 (miniconda etc.)
- Visual Studio >= 2019 (for Windows)
git clone --recursive https://github.com/DwangoMediaVillage/pydomino
cd pydomino
pip install ./
また、下記のように直接 pip インストールもできます(コマンドラインツールもインストールされます。コマンドラインツールを使うときは、onnxファイルを指定する必要があります。onnx_model/phoneme_transition_model.onnx
をダウンロードしてください):
pip install git+https://github.com/DwangoMediaVillage/pydomino
Anaconda Prompt (miniconda3)
環境において MSVC の vcvars64.bat
を利用してインストールします:
"C:\Program Files (x86)\Microsoft Visual Studio\2019\Professional\VC\Auxiliary\Build\vcvars64.bat"
or"C:\Program Files\Microsoft Visual Studio\2022\Community\VC\Auxiliary\Build\vcvars64.bat"
.
# on `Anaconda Prompt (miniconda3)`
"C:\Program Files\Microsoft Visual Studio\2022\Community\VC\Auxiliary\Build\vcvars64.bat"
git clone --recursive https://github.com/DwangoMediaVillage/pydomino
cd pydomino
pip install ./
また、下記のように直接 pip インストールもできます(コマンドラインツールもインストールされます。コマンドラインツールを使うときは、onnxファイルを指定する必要があります。onnx_model/phoneme_transition_model.onnx
をダウンロードしてください):
pip install git+https://github.com/DwangoMediaVillage/pydomino
alignmer: pydomino.Aligner = pydomino.Aligner(path-to-model-file.onnx)
y: np.ndarray = librosa.load(path-to-wav-file, sr=16_000, mono=True, dtype=np.float32)[0]
p: list[str] = path-to-phoneme-file.read_text().split(" ")
z: list[tuple[float, float, str]] = alignmer.align(y, " ".join(p), 3) # [(start_time_sec, end_time_sec, phoneme_str)]
path-to-model-file.onnx
は事前学習済みの onnx モデルファイルです。onnx_model/phoneme_transition_model.onnx
にあります。
path-to-wav-file
はサンプリング周波数 16kHz のモノラル wav ファイルです。path-to-phoneme-file
は音素を空白区切りしたテキストが格納されたファイルのパスです。- NOTE: 開始音素と終了音素は
pau
である必要があります。
- NOTE: 開始音素と終了音素は
phonemes
に使える音素一覧は下記の通りです:
pau |
ry |
r |
my |
m |
ny |
n |
j |
z |
by |
b |
dy |
d |
gy |
g |
ky |
k |
ch |
ts |
sh |
s |
hy |
h |
v |
f |
py |
p |
t |
y |
w |
N |
a |
i |
u |
e |
o |
I |
U |
cl |
上記のインストール手順における pip install
により Cli ツールも自動でビルドされます。
ビルドされたツールは下記のようにして使えます:
domino \
--input_path={path-to-wav-file} \
--input_phoneme={path-to-phoneme-file} \
--output_path={path-to-output-lab-file} \
--onnx_path={path-to-output-onnx-file} \
--min_frame==3
onnxファイルは当組織で学習済みの onnx_model/phoneme_transition_model.onnx
を用意していますのでお使いください
アラインメント結果のラベルファイル (.lab) は、tsv ファイル構造になっています。
各行に音素の開始時刻と終了時刻 (いずれも単位は秒) と、そのときの音素が TAB 区切りで並んでいます:
0.000 0.110 pau
0.110 0.140 d
0.140 0.170 o
0.170 0.210 w
0.210 0.360 a
0.360 0.450 N
0.450 0.490 g
0.490 0.620 o
0.620 0.755 pau