Spaces:
Running
Running
TheStinger
commited on
Commit
•
18bba3b
1
Parent(s):
33685d0
Upload 36 files
Browse files- README.md +57 -0
- app.py +2088 -0
- audios/Test_Audio.mp3 +0 -0
- audios/gitignore.txt +0 -0
- config.py +204 -0
- gitattributes.txt +35 -0
- gitignore.txt +12 -0
- i18n.py +28 -0
- i18n/en_US.json +130 -0
- i18n/es_ES.json +132 -0
- i18n/it_IT.json +130 -0
- i18n/ja_JP.json +126 -0
- i18n/locale_diff.py +45 -0
- i18n/ru-RU.json +130 -0
- i18n/tr_TR.json +130 -0
- i18n/zh_CN.json +132 -0
- i18n/zh_HK.json +132 -0
- i18n/zh_SG.json +132 -0
- i18n/zh_TW.json +132 -0
- lib/infer_pack/attentions.py +417 -0
- lib/infer_pack/commons.py +166 -0
- lib/infer_pack/models.py +1142 -0
- lib/infer_pack/models_onnx.py +819 -0
- lib/infer_pack/modules.py +522 -0
- lib/infer_pack/modules/F0Predictor/DioF0Predictor.py +90 -0
- lib/infer_pack/modules/F0Predictor/F0Predictor.py +16 -0
- lib/infer_pack/modules/F0Predictor/HarvestF0Predictor.py +86 -0
- lib/infer_pack/modules/F0Predictor/PMF0Predictor.py +97 -0
- lib/infer_pack/modules/F0Predictor/__init__.py +0 -0
- lib/infer_pack/transforms.py +209 -0
- packages.txt +3 -0
- requirements.txt +22 -0
- rmvpe.py +432 -0
- run.sh +16 -0
- utils.py +151 -0
- vc_infer_pipeline.py +646 -0
README.md
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
title: Ilaria RVC 2.0
|
3 |
+
emoji: 😻
|
4 |
+
colorFrom: pink
|
5 |
+
colorTo: pink
|
6 |
+
sdk: gradio
|
7 |
+
sdk_version: 3.42.0
|
8 |
+
app_file: app.py
|
9 |
+
pinned: true
|
10 |
+
license: lgpl-3.0
|
11 |
+
---
|
12 |
+
|
13 |
+
## 🔧 Pre-requisites
|
14 |
+
|
15 |
+
Before running the project, you must have the following tool installed on your machine:
|
16 |
+
* [Python v3.8.0](https://www.python.org/downloads/release/python-380/)
|
17 |
+
|
18 |
+
Also, you will need to clone the repository:
|
19 |
+
|
20 |
+
```bash
|
21 |
+
# Clone the repository
|
22 |
+
git clone https://huggingface.co/spaces/mateuseap/magic-vocals/
|
23 |
+
# Enter in the root directory
|
24 |
+
cd magic-vocals
|
25 |
+
```
|
26 |
+
|
27 |
+
## 🚀 How to run
|
28 |
+
|
29 |
+
After you've cloned the repository and entered in the root directory, run the following commands:
|
30 |
+
|
31 |
+
```bash
|
32 |
+
# Create and activate a Virtual Environment (make sure you're using Python v3.8.0 to do it)
|
33 |
+
python -m venv venv
|
34 |
+
. venv/bin/activate
|
35 |
+
|
36 |
+
# Change mode and execute a shell script to configure and run the application
|
37 |
+
chmod +x run.sh
|
38 |
+
./run.sh
|
39 |
+
```
|
40 |
+
|
41 |
+
After the shell script executes everything, the application will be running at http://127.0.0.1:7860! Open up the link in a browser to use the app:
|
42 |
+
|
43 |
+
![Magic Vocals](https://i.imgur.com/V55oKv8.png)
|
44 |
+
|
45 |
+
**You only need to execute the `run.sh` one time**, once you've executed it one time, you just need to activate the virtual environment and run the command below to start the app again:
|
46 |
+
|
47 |
+
```bash
|
48 |
+
python app.py
|
49 |
+
```
|
50 |
+
|
51 |
+
**THE `run.sh` IS SUPPORTED BY THE FOLLOWING OPERATING SYSTEMS:**
|
52 |
+
|
53 |
+
|
54 |
+
| OS | Supported |
|
55 |
+
|-----------|:---------:|
|
56 |
+
| `Windows` | ❌ |
|
57 |
+
| `Ubuntu` | ✅ |
|
app.py
ADDED
@@ -0,0 +1,2088 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import subprocess, torch, os, traceback, sys, warnings, shutil, numpy as np
|
2 |
+
from mega import Mega
|
3 |
+
os.environ["no_proxy"] = "localhost, 127.0.0.1, ::1"
|
4 |
+
import threading
|
5 |
+
from time import sleep
|
6 |
+
from subprocess import Popen
|
7 |
+
import faiss
|
8 |
+
from random import shuffle
|
9 |
+
import json, datetime, requests
|
10 |
+
from gtts import gTTS
|
11 |
+
now_dir = os.getcwd()
|
12 |
+
sys.path.append(now_dir)
|
13 |
+
tmp = os.path.join(now_dir, "TEMP")
|
14 |
+
shutil.rmtree(tmp, ignore_errors=True)
|
15 |
+
shutil.rmtree("%s/runtime/Lib/site-packages/infer_pack" % (now_dir), ignore_errors=True)
|
16 |
+
os.makedirs(tmp, exist_ok=True)
|
17 |
+
os.makedirs(os.path.join(now_dir, "logs"), exist_ok=True)
|
18 |
+
os.makedirs(os.path.join(now_dir, "weights"), exist_ok=True)
|
19 |
+
os.environ["TEMP"] = tmp
|
20 |
+
warnings.filterwarnings("ignore")
|
21 |
+
torch.manual_seed(114514)
|
22 |
+
from i18n import I18nAuto
|
23 |
+
|
24 |
+
import signal
|
25 |
+
|
26 |
+
import math
|
27 |
+
|
28 |
+
from utils import load_audio, CSVutil
|
29 |
+
|
30 |
+
global DoFormant, Quefrency, Timbre
|
31 |
+
|
32 |
+
if not os.path.isdir('csvdb/'):
|
33 |
+
os.makedirs('csvdb')
|
34 |
+
frmnt, stp = open("csvdb/formanting.csv", 'w'), open("csvdb/stop.csv", 'w')
|
35 |
+
frmnt.close()
|
36 |
+
stp.close()
|
37 |
+
|
38 |
+
try:
|
39 |
+
DoFormant, Quefrency, Timbre = CSVutil('csvdb/formanting.csv', 'r', 'formanting')
|
40 |
+
DoFormant = (
|
41 |
+
lambda DoFormant: True if DoFormant.lower() == 'true' else (False if DoFormant.lower() == 'false' else DoFormant)
|
42 |
+
)(DoFormant)
|
43 |
+
except (ValueError, TypeError, IndexError):
|
44 |
+
DoFormant, Quefrency, Timbre = False, 1.0, 1.0
|
45 |
+
CSVutil('csvdb/formanting.csv', 'w+', 'formanting', DoFormant, Quefrency, Timbre)
|
46 |
+
|
47 |
+
def download_models():
|
48 |
+
# Download hubert base model if not present
|
49 |
+
if not os.path.isfile('./hubert_base.pt'):
|
50 |
+
response = requests.get('https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/hubert_base.pt')
|
51 |
+
|
52 |
+
if response.status_code == 200:
|
53 |
+
with open('./hubert_base.pt', 'wb') as f:
|
54 |
+
f.write(response.content)
|
55 |
+
print("Downloaded hubert base model file successfully. File saved to ./hubert_base.pt.")
|
56 |
+
else:
|
57 |
+
raise Exception("Failed to download hubert base model file. Status code: " + str(response.status_code) + ".")
|
58 |
+
|
59 |
+
# Download rmvpe model if not present
|
60 |
+
if not os.path.isfile('./rmvpe.pt'):
|
61 |
+
response = requests.get('https://drive.usercontent.google.com/download?id=1Hkn4kNuVFRCNQwyxQFRtmzmMBGpQxptI&export=download&authuser=0&confirm=t&uuid=0b3a40de-465b-4c65-8c41-135b0b45c3f7&at=APZUnTV3lA3LnyTbeuduura6Dmi2:1693724254058')
|
62 |
+
|
63 |
+
if response.status_code == 200:
|
64 |
+
with open('./rmvpe.pt', 'wb') as f:
|
65 |
+
f.write(response.content)
|
66 |
+
print("Downloaded rmvpe model file successfully. File saved to ./rmvpe.pt.")
|
67 |
+
else:
|
68 |
+
raise Exception("Failed to download rmvpe model file. Status code: " + str(response.status_code) + ".")
|
69 |
+
|
70 |
+
download_models()
|
71 |
+
|
72 |
+
print("\n-------------------------------\nRVC v2 Easy GUI (Local Edition)\n-------------------------------\n")
|
73 |
+
|
74 |
+
def formant_apply(qfrency, tmbre):
|
75 |
+
Quefrency = qfrency
|
76 |
+
Timbre = tmbre
|
77 |
+
DoFormant = True
|
78 |
+
CSVutil('csvdb/formanting.csv', 'w+', 'formanting', DoFormant, qfrency, tmbre)
|
79 |
+
|
80 |
+
return ({"value": Quefrency, "__type__": "update"}, {"value": Timbre, "__type__": "update"})
|
81 |
+
|
82 |
+
def get_fshift_presets():
|
83 |
+
fshift_presets_list = []
|
84 |
+
for dirpath, _, filenames in os.walk("./formantshiftcfg/"):
|
85 |
+
for filename in filenames:
|
86 |
+
if filename.endswith(".txt"):
|
87 |
+
fshift_presets_list.append(os.path.join(dirpath,filename).replace('\\','/'))
|
88 |
+
|
89 |
+
if len(fshift_presets_list) > 0:
|
90 |
+
return fshift_presets_list
|
91 |
+
else:
|
92 |
+
return ''
|
93 |
+
|
94 |
+
|
95 |
+
|
96 |
+
def formant_enabled(cbox, qfrency, tmbre, frmntapply, formantpreset, formant_refresh_button):
|
97 |
+
|
98 |
+
if (cbox):
|
99 |
+
|
100 |
+
DoFormant = True
|
101 |
+
CSVutil('csvdb/formanting.csv', 'w+', 'formanting', DoFormant, qfrency, tmbre)
|
102 |
+
#print(f"is checked? - {cbox}\ngot {DoFormant}")
|
103 |
+
|
104 |
+
return (
|
105 |
+
{"value": True, "__type__": "update"},
|
106 |
+
{"visible": True, "__type__": "update"},
|
107 |
+
{"visible": True, "__type__": "update"},
|
108 |
+
{"visible": True, "__type__": "update"},
|
109 |
+
{"visible": True, "__type__": "update"},
|
110 |
+
{"visible": True, "__type__": "update"},
|
111 |
+
)
|
112 |
+
|
113 |
+
|
114 |
+
else:
|
115 |
+
|
116 |
+
DoFormant = False
|
117 |
+
CSVutil('csvdb/formanting.csv', 'w+', 'formanting', DoFormant, qfrency, tmbre)
|
118 |
+
|
119 |
+
#print(f"is checked? - {cbox}\ngot {DoFormant}")
|
120 |
+
return (
|
121 |
+
{"value": False, "__type__": "update"},
|
122 |
+
{"visible": False, "__type__": "update"},
|
123 |
+
{"visible": False, "__type__": "update"},
|
124 |
+
{"visible": False, "__type__": "update"},
|
125 |
+
{"visible": False, "__type__": "update"},
|
126 |
+
{"visible": False, "__type__": "update"},
|
127 |
+
{"visible": False, "__type__": "update"},
|
128 |
+
)
|
129 |
+
|
130 |
+
|
131 |
+
|
132 |
+
def preset_apply(preset, qfer, tmbr):
|
133 |
+
if str(preset) != '':
|
134 |
+
with open(str(preset), 'r') as p:
|
135 |
+
content = p.readlines()
|
136 |
+
qfer, tmbr = content[0].split('\n')[0], content[1]
|
137 |
+
|
138 |
+
formant_apply(qfer, tmbr)
|
139 |
+
else:
|
140 |
+
pass
|
141 |
+
return ({"value": qfer, "__type__": "update"}, {"value": tmbr, "__type__": "update"})
|
142 |
+
|
143 |
+
def update_fshift_presets(preset, qfrency, tmbre):
|
144 |
+
|
145 |
+
qfrency, tmbre = preset_apply(preset, qfrency, tmbre)
|
146 |
+
|
147 |
+
if (str(preset) != ''):
|
148 |
+
with open(str(preset), 'r') as p:
|
149 |
+
content = p.readlines()
|
150 |
+
qfrency, tmbre = content[0].split('\n')[0], content[1]
|
151 |
+
|
152 |
+
formant_apply(qfrency, tmbre)
|
153 |
+
else:
|
154 |
+
pass
|
155 |
+
return (
|
156 |
+
{"choices": get_fshift_presets(), "__type__": "update"},
|
157 |
+
{"value": qfrency, "__type__": "update"},
|
158 |
+
{"value": tmbre, "__type__": "update"},
|
159 |
+
)
|
160 |
+
|
161 |
+
i18n = I18nAuto()
|
162 |
+
#i18n.print()
|
163 |
+
# 判断是否有能用来训练和加速推理的N卡
|
164 |
+
ngpu = torch.cuda.device_count()
|
165 |
+
gpu_infos = []
|
166 |
+
mem = []
|
167 |
+
if (not torch.cuda.is_available()) or ngpu == 0:
|
168 |
+
if_gpu_ok = False
|
169 |
+
else:
|
170 |
+
if_gpu_ok = False
|
171 |
+
for i in range(ngpu):
|
172 |
+
gpu_name = torch.cuda.get_device_name(i)
|
173 |
+
if (
|
174 |
+
"10" in gpu_name
|
175 |
+
or "16" in gpu_name
|
176 |
+
or "20" in gpu_name
|
177 |
+
or "30" in gpu_name
|
178 |
+
or "40" in gpu_name
|
179 |
+
or "A2" in gpu_name.upper()
|
180 |
+
or "A3" in gpu_name.upper()
|
181 |
+
or "A4" in gpu_name.upper()
|
182 |
+
or "P4" in gpu_name.upper()
|
183 |
+
or "A50" in gpu_name.upper()
|
184 |
+
or "A60" in gpu_name.upper()
|
185 |
+
or "70" in gpu_name
|
186 |
+
or "80" in gpu_name
|
187 |
+
or "90" in gpu_name
|
188 |
+
or "M4" in gpu_name.upper()
|
189 |
+
or "T4" in gpu_name.upper()
|
190 |
+
or "TITAN" in gpu_name.upper()
|
191 |
+
): # A10#A100#V100#A40#P40#M40#K80#A4500
|
192 |
+
if_gpu_ok = True # 至少有一张能用的N卡
|
193 |
+
gpu_infos.append("%s\t%s" % (i, gpu_name))
|
194 |
+
mem.append(
|
195 |
+
int(
|
196 |
+
torch.cuda.get_device_properties(i).total_memory
|
197 |
+
/ 1024
|
198 |
+
/ 1024
|
199 |
+
/ 1024
|
200 |
+
+ 0.4
|
201 |
+
)
|
202 |
+
)
|
203 |
+
if if_gpu_ok == True and len(gpu_infos) > 0:
|
204 |
+
gpu_info = "\n".join(gpu_infos)
|
205 |
+
default_batch_size = min(mem) // 2
|
206 |
+
else:
|
207 |
+
gpu_info = i18n("很遗憾您这没有能用的显卡来支持您训练")
|
208 |
+
default_batch_size = 1
|
209 |
+
gpus = "-".join([i[0] for i in gpu_infos])
|
210 |
+
from lib.infer_pack.models import (
|
211 |
+
SynthesizerTrnMs256NSFsid,
|
212 |
+
SynthesizerTrnMs256NSFsid_nono,
|
213 |
+
SynthesizerTrnMs768NSFsid,
|
214 |
+
SynthesizerTrnMs768NSFsid_nono,
|
215 |
+
)
|
216 |
+
import soundfile as sf
|
217 |
+
from fairseq import checkpoint_utils
|
218 |
+
import gradio as gr
|
219 |
+
import logging
|
220 |
+
from vc_infer_pipeline import VC
|
221 |
+
from config import Config
|
222 |
+
|
223 |
+
config = Config()
|
224 |
+
# from trainset_preprocess_pipeline import PreProcess
|
225 |
+
logging.getLogger("numba").setLevel(logging.WARNING)
|
226 |
+
|
227 |
+
hubert_model = None
|
228 |
+
|
229 |
+
def load_hubert():
|
230 |
+
global hubert_model
|
231 |
+
models, _, _ = checkpoint_utils.load_model_ensemble_and_task(
|
232 |
+
["hubert_base.pt"],
|
233 |
+
suffix="",
|
234 |
+
)
|
235 |
+
hubert_model = models[0]
|
236 |
+
hubert_model = hubert_model.to(config.device)
|
237 |
+
if config.is_half:
|
238 |
+
hubert_model = hubert_model.half()
|
239 |
+
else:
|
240 |
+
hubert_model = hubert_model.float()
|
241 |
+
hubert_model.eval()
|
242 |
+
|
243 |
+
|
244 |
+
weight_root = "weights"
|
245 |
+
index_root = "logs"
|
246 |
+
names = []
|
247 |
+
for name in os.listdir(weight_root):
|
248 |
+
if name.endswith(".pth"):
|
249 |
+
names.append(name)
|
250 |
+
index_paths = []
|
251 |
+
for root, dirs, files in os.walk(index_root, topdown=False):
|
252 |
+
for name in files:
|
253 |
+
if name.endswith(".index") and "trained" not in name:
|
254 |
+
index_paths.append("%s/%s" % (root, name))
|
255 |
+
|
256 |
+
|
257 |
+
|
258 |
+
def vc_single(
|
259 |
+
sid,
|
260 |
+
input_audio_path,
|
261 |
+
f0_up_key,
|
262 |
+
f0_file,
|
263 |
+
f0_method,
|
264 |
+
file_index,
|
265 |
+
#file_index2,
|
266 |
+
# file_big_npy,
|
267 |
+
index_rate,
|
268 |
+
filter_radius,
|
269 |
+
resample_sr,
|
270 |
+
rms_mix_rate,
|
271 |
+
protect,
|
272 |
+
crepe_hop_length,
|
273 |
+
): # spk_item, input_audio0, vc_transform0,f0_file,f0method0
|
274 |
+
global tgt_sr, net_g, vc, hubert_model, version
|
275 |
+
if input_audio_path is None:
|
276 |
+
return "You need to upload an audio", None
|
277 |
+
f0_up_key = int(f0_up_key)
|
278 |
+
try:
|
279 |
+
audio = load_audio(input_audio_path, 16000, DoFormant, Quefrency, Timbre)
|
280 |
+
audio_max = np.abs(audio).max() / 0.95
|
281 |
+
if audio_max > 1:
|
282 |
+
audio /= audio_max
|
283 |
+
times = [0, 0, 0]
|
284 |
+
if hubert_model == None:
|
285 |
+
load_hubert()
|
286 |
+
if_f0 = cpt.get("f0", 1)
|
287 |
+
file_index = (
|
288 |
+
(
|
289 |
+
file_index.strip(" ")
|
290 |
+
.strip('"')
|
291 |
+
.strip("\n")
|
292 |
+
.strip('"')
|
293 |
+
.strip(" ")
|
294 |
+
.replace("trained", "added")
|
295 |
+
)
|
296 |
+
) # 防止小白写错,自动帮他替换掉
|
297 |
+
# file_big_npy = (
|
298 |
+
# file_big_npy.strip(" ").strip('"').strip("\n").strip('"').strip(" ")
|
299 |
+
# )
|
300 |
+
audio_opt = vc.pipeline(
|
301 |
+
hubert_model,
|
302 |
+
net_g,
|
303 |
+
sid,
|
304 |
+
audio,
|
305 |
+
input_audio_path,
|
306 |
+
times,
|
307 |
+
f0_up_key,
|
308 |
+
f0_method,
|
309 |
+
file_index,
|
310 |
+
# file_big_npy,
|
311 |
+
index_rate,
|
312 |
+
if_f0,
|
313 |
+
filter_radius,
|
314 |
+
tgt_sr,
|
315 |
+
resample_sr,
|
316 |
+
rms_mix_rate,
|
317 |
+
version,
|
318 |
+
protect,
|
319 |
+
crepe_hop_length,
|
320 |
+
f0_file=f0_file,
|
321 |
+
)
|
322 |
+
if resample_sr >= 16000 and tgt_sr != resample_sr:
|
323 |
+
tgt_sr = resample_sr
|
324 |
+
index_info = (
|
325 |
+
"Using index:%s." % file_index
|
326 |
+
if os.path.exists(file_index)
|
327 |
+
else "Index not used."
|
328 |
+
)
|
329 |
+
return "Success.\n %s\nTime:\n npy:%ss, f0:%ss, infer:%ss" % (
|
330 |
+
index_info,
|
331 |
+
times[0],
|
332 |
+
times[1],
|
333 |
+
times[2],
|
334 |
+
), (tgt_sr, audio_opt)
|
335 |
+
except:
|
336 |
+
info = traceback.format_exc()
|
337 |
+
print(info)
|
338 |
+
return info, (None, None)
|
339 |
+
|
340 |
+
|
341 |
+
def vc_multi(
|
342 |
+
sid,
|
343 |
+
dir_path,
|
344 |
+
opt_root,
|
345 |
+
paths,
|
346 |
+
f0_up_key,
|
347 |
+
f0_method,
|
348 |
+
file_index,
|
349 |
+
file_index2,
|
350 |
+
# file_big_npy,
|
351 |
+
index_rate,
|
352 |
+
filter_radius,
|
353 |
+
resample_sr,
|
354 |
+
rms_mix_rate,
|
355 |
+
protect,
|
356 |
+
format1,
|
357 |
+
crepe_hop_length,
|
358 |
+
):
|
359 |
+
try:
|
360 |
+
dir_path = (
|
361 |
+
dir_path.strip(" ").strip('"').strip("\n").strip('"').strip(" ")
|
362 |
+
) # 防止小白拷路径头尾带了空格和"和回车
|
363 |
+
opt_root = opt_root.strip(" ").strip('"').strip("\n").strip('"').strip(" ")
|
364 |
+
os.makedirs(opt_root, exist_ok=True)
|
365 |
+
try:
|
366 |
+
if dir_path != "":
|
367 |
+
paths = [os.path.join(dir_path, name) for name in os.listdir(dir_path)]
|
368 |
+
else:
|
369 |
+
paths = [path.name for path in paths]
|
370 |
+
except:
|
371 |
+
traceback.print_exc()
|
372 |
+
paths = [path.name for path in paths]
|
373 |
+
infos = []
|
374 |
+
for path in paths:
|
375 |
+
info, opt = vc_single(
|
376 |
+
sid,
|
377 |
+
path,
|
378 |
+
f0_up_key,
|
379 |
+
None,
|
380 |
+
f0_method,
|
381 |
+
file_index,
|
382 |
+
# file_big_npy,
|
383 |
+
index_rate,
|
384 |
+
filter_radius,
|
385 |
+
resample_sr,
|
386 |
+
rms_mix_rate,
|
387 |
+
protect,
|
388 |
+
crepe_hop_length
|
389 |
+
)
|
390 |
+
if "Success" in info:
|
391 |
+
try:
|
392 |
+
tgt_sr, audio_opt = opt
|
393 |
+
if format1 in ["wav", "flac"]:
|
394 |
+
sf.write(
|
395 |
+
"%s/%s.%s" % (opt_root, os.path.basename(path), format1),
|
396 |
+
audio_opt,
|
397 |
+
tgt_sr,
|
398 |
+
)
|
399 |
+
else:
|
400 |
+
path = "%s/%s.wav" % (opt_root, os.path.basename(path))
|
401 |
+
sf.write(
|
402 |
+
path,
|
403 |
+
audio_opt,
|
404 |
+
tgt_sr,
|
405 |
+
)
|
406 |
+
if os.path.exists(path):
|
407 |
+
os.system(
|
408 |
+
"ffmpeg -i %s -vn %s -q:a 2 -y"
|
409 |
+
% (path, path[:-4] + ".%s" % format1)
|
410 |
+
)
|
411 |
+
except:
|
412 |
+
info += traceback.format_exc()
|
413 |
+
infos.append("%s->%s" % (os.path.basename(path), info))
|
414 |
+
yield "\n".join(infos)
|
415 |
+
yield "\n".join(infos)
|
416 |
+
except:
|
417 |
+
yield traceback.format_exc()
|
418 |
+
|
419 |
+
# 一个选项卡全局只能有一个音色
|
420 |
+
def get_vc(sid):
|
421 |
+
global n_spk, tgt_sr, net_g, vc, cpt, version
|
422 |
+
if sid == "" or sid == []:
|
423 |
+
global hubert_model
|
424 |
+
if hubert_model != None: # 考虑到轮询, 需要加个判断看是否 sid 是由有模型切换到无模型的
|
425 |
+
print("clean_empty_cache")
|
426 |
+
del net_g, n_spk, vc, hubert_model, tgt_sr # ,cpt
|
427 |
+
hubert_model = net_g = n_spk = vc = hubert_model = tgt_sr = None
|
428 |
+
if torch.cuda.is_available():
|
429 |
+
torch.cuda.empty_cache()
|
430 |
+
###楼下不这么折腾清理不干净
|
431 |
+
if_f0 = cpt.get("f0", 1)
|
432 |
+
version = cpt.get("version", "v1")
|
433 |
+
if version == "v1":
|
434 |
+
if if_f0 == 1:
|
435 |
+
net_g = SynthesizerTrnMs256NSFsid(
|
436 |
+
*cpt["config"], is_half=config.is_half
|
437 |
+
)
|
438 |
+
else:
|
439 |
+
net_g = SynthesizerTrnMs256NSFsid_nono(*cpt["config"])
|
440 |
+
elif version == "v2":
|
441 |
+
if if_f0 == 1:
|
442 |
+
net_g = SynthesizerTrnMs768NSFsid(
|
443 |
+
*cpt["config"], is_half=config.is_half
|
444 |
+
)
|
445 |
+
else:
|
446 |
+
net_g = SynthesizerTrnMs768NSFsid_nono(*cpt["config"])
|
447 |
+
del net_g, cpt
|
448 |
+
if torch.cuda.is_available():
|
449 |
+
torch.cuda.empty_cache()
|
450 |
+
cpt = None
|
451 |
+
return {"visible": False, "__type__": "update"}
|
452 |
+
person = "%s/%s" % (weight_root, sid)
|
453 |
+
print("loading %s" % person)
|
454 |
+
cpt = torch.load(person, map_location="cpu")
|
455 |
+
tgt_sr = cpt["config"][-1]
|
456 |
+
cpt["config"][-3] = cpt["weight"]["emb_g.weight"].shape[0] # n_spk
|
457 |
+
if_f0 = cpt.get("f0", 1)
|
458 |
+
version = cpt.get("version", "v1")
|
459 |
+
if version == "v1":
|
460 |
+
if if_f0 == 1:
|
461 |
+
net_g = SynthesizerTrnMs256NSFsid(*cpt["config"], is_half=config.is_half)
|
462 |
+
else:
|
463 |
+
net_g = SynthesizerTrnMs256NSFsid_nono(*cpt["config"])
|
464 |
+
elif version == "v2":
|
465 |
+
if if_f0 == 1:
|
466 |
+
net_g = SynthesizerTrnMs768NSFsid(*cpt["config"], is_half=config.is_half)
|
467 |
+
else:
|
468 |
+
net_g = SynthesizerTrnMs768NSFsid_nono(*cpt["config"])
|
469 |
+
del net_g.enc_q
|
470 |
+
print(net_g.load_state_dict(cpt["weight"], strict=False))
|
471 |
+
net_g.eval().to(config.device)
|
472 |
+
if config.is_half:
|
473 |
+
net_g = net_g.half()
|
474 |
+
else:
|
475 |
+
net_g = net_g.float()
|
476 |
+
vc = VC(tgt_sr, config)
|
477 |
+
n_spk = cpt["config"][-3]
|
478 |
+
return {"visible": False, "maximum": n_spk, "__type__": "update"}
|
479 |
+
|
480 |
+
|
481 |
+
def change_choices():
|
482 |
+
names = []
|
483 |
+
for name in os.listdir(weight_root):
|
484 |
+
if name.endswith(".pth"):
|
485 |
+
names.append(name)
|
486 |
+
index_paths = []
|
487 |
+
for root, dirs, files in os.walk(index_root, topdown=False):
|
488 |
+
for name in files:
|
489 |
+
if name.endswith(".index") and "trained" not in name:
|
490 |
+
index_paths.append("%s/%s" % (root, name))
|
491 |
+
return {"choices": sorted(names), "__type__": "update"}, {
|
492 |
+
"choices": sorted(index_paths),
|
493 |
+
"__type__": "update",
|
494 |
+
}
|
495 |
+
|
496 |
+
|
497 |
+
def clean():
|
498 |
+
return {"value": "", "__type__": "update"}
|
499 |
+
|
500 |
+
|
501 |
+
sr_dict = {
|
502 |
+
"32k": 32000,
|
503 |
+
"40k": 40000,
|
504 |
+
"48k": 48000,
|
505 |
+
}
|
506 |
+
|
507 |
+
|
508 |
+
def if_done(done, p):
|
509 |
+
while 1:
|
510 |
+
if p.poll() == None:
|
511 |
+
sleep(0.5)
|
512 |
+
else:
|
513 |
+
break
|
514 |
+
done[0] = True
|
515 |
+
|
516 |
+
|
517 |
+
def if_done_multi(done, ps):
|
518 |
+
while 1:
|
519 |
+
# poll==None代表进程未结束
|
520 |
+
# 只要有一个进程未结束都不停
|
521 |
+
flag = 1
|
522 |
+
for p in ps:
|
523 |
+
if p.poll() == None:
|
524 |
+
flag = 0
|
525 |
+
sleep(0.5)
|
526 |
+
break
|
527 |
+
if flag == 1:
|
528 |
+
break
|
529 |
+
done[0] = True
|
530 |
+
|
531 |
+
|
532 |
+
def preprocess_dataset(trainset_dir, exp_dir, sr, n_p):
|
533 |
+
sr = sr_dict[sr]
|
534 |
+
os.makedirs("%s/logs/%s" % (now_dir, exp_dir), exist_ok=True)
|
535 |
+
f = open("%s/logs/%s/preprocess.log" % (now_dir, exp_dir), "w")
|
536 |
+
f.close()
|
537 |
+
cmd = (
|
538 |
+
config.python_cmd
|
539 |
+
+ " trainset_preprocess_pipeline_print.py %s %s %s %s/logs/%s "
|
540 |
+
% (trainset_dir, sr, n_p, now_dir, exp_dir)
|
541 |
+
+ str(config.noparallel)
|
542 |
+
)
|
543 |
+
print(cmd)
|
544 |
+
p = Popen(cmd, shell=True) # , stdin=PIPE, stdout=PIPE,stderr=PIPE,cwd=now_dir
|
545 |
+
###煞笔gr, popen read都非得全跑完了再一次性读取, 不用gr就正常读一句输出一句;只能额外弄出一个文本流定时读
|
546 |
+
done = [False]
|
547 |
+
threading.Thread(
|
548 |
+
target=if_done,
|
549 |
+
args=(
|
550 |
+
done,
|
551 |
+
p,
|
552 |
+
),
|
553 |
+
).start()
|
554 |
+
while 1:
|
555 |
+
with open("%s/logs/%s/preprocess.log" % (now_dir, exp_dir), "r") as f:
|
556 |
+
yield (f.read())
|
557 |
+
sleep(1)
|
558 |
+
if done[0] == True:
|
559 |
+
break
|
560 |
+
with open("%s/logs/%s/preprocess.log" % (now_dir, exp_dir), "r") as f:
|
561 |
+
log = f.read()
|
562 |
+
print(log)
|
563 |
+
yield log
|
564 |
+
|
565 |
+
# but2.click(extract_f0,[gpus6,np7,f0method8,if_f0_3,trainset_dir4],[info2])
|
566 |
+
def extract_f0_feature(gpus, n_p, f0method, if_f0, exp_dir, version19, echl):
|
567 |
+
gpus = gpus.split("-")
|
568 |
+
os.makedirs("%s/logs/%s" % (now_dir, exp_dir), exist_ok=True)
|
569 |
+
f = open("%s/logs/%s/extract_f0_feature.log" % (now_dir, exp_dir), "w")
|
570 |
+
f.close()
|
571 |
+
if if_f0:
|
572 |
+
cmd = config.python_cmd + " extract_f0_print.py %s/logs/%s %s %s %s" % (
|
573 |
+
now_dir,
|
574 |
+
exp_dir,
|
575 |
+
n_p,
|
576 |
+
f0method,
|
577 |
+
echl,
|
578 |
+
)
|
579 |
+
print(cmd)
|
580 |
+
p = Popen(cmd, shell=True, cwd=now_dir) # , stdin=PIPE, stdout=PIPE,stderr=PIPE
|
581 |
+
###煞笔gr, popen read都非得全跑完了再一次性读取, 不用gr就正常读一句输出一句;只能额外弄出一个文本流定时读
|
582 |
+
done = [False]
|
583 |
+
threading.Thread(
|
584 |
+
target=if_done,
|
585 |
+
args=(
|
586 |
+
done,
|
587 |
+
p,
|
588 |
+
),
|
589 |
+
).start()
|
590 |
+
while 1:
|
591 |
+
with open(
|
592 |
+
"%s/logs/%s/extract_f0_feature.log" % (now_dir, exp_dir), "r"
|
593 |
+
) as f:
|
594 |
+
yield (f.read())
|
595 |
+
sleep(1)
|
596 |
+
if done[0] == True:
|
597 |
+
break
|
598 |
+
with open("%s/logs/%s/extract_f0_feature.log" % (now_dir, exp_dir), "r") as f:
|
599 |
+
log = f.read()
|
600 |
+
print(log)
|
601 |
+
yield log
|
602 |
+
####对不同part分别开多进程
|
603 |
+
"""
|
604 |
+
n_part=int(sys.argv[1])
|
605 |
+
i_part=int(sys.argv[2])
|
606 |
+
i_gpu=sys.argv[3]
|
607 |
+
exp_dir=sys.argv[4]
|
608 |
+
os.environ["CUDA_VISIBLE_DEVICES"]=str(i_gpu)
|
609 |
+
"""
|
610 |
+
leng = len(gpus)
|
611 |
+
ps = []
|
612 |
+
for idx, n_g in enumerate(gpus):
|
613 |
+
cmd = (
|
614 |
+
config.python_cmd
|
615 |
+
+ " extract_feature_print.py %s %s %s %s %s/logs/%s %s"
|
616 |
+
% (
|
617 |
+
config.device,
|
618 |
+
leng,
|
619 |
+
idx,
|
620 |
+
n_g,
|
621 |
+
now_dir,
|
622 |
+
exp_dir,
|
623 |
+
version19,
|
624 |
+
)
|
625 |
+
)
|
626 |
+
print(cmd)
|
627 |
+
p = Popen(
|
628 |
+
cmd, shell=True, cwd=now_dir
|
629 |
+
) # , shell=True, stdin=PIPE, stdout=PIPE, stderr=PIPE, cwd=now_dir
|
630 |
+
ps.append(p)
|
631 |
+
###煞笔gr, popen read都非得全跑完了再一次性读取, 不用gr就正常读一句输出一句;只能额外弄出一个文本流定时读
|
632 |
+
done = [False]
|
633 |
+
threading.Thread(
|
634 |
+
target=if_done_multi,
|
635 |
+
args=(
|
636 |
+
done,
|
637 |
+
ps,
|
638 |
+
),
|
639 |
+
).start()
|
640 |
+
while 1:
|
641 |
+
with open("%s/logs/%s/extract_f0_feature.log" % (now_dir, exp_dir), "r") as f:
|
642 |
+
yield (f.read())
|
643 |
+
sleep(1)
|
644 |
+
if done[0] == True:
|
645 |
+
break
|
646 |
+
with open("%s/logs/%s/extract_f0_feature.log" % (now_dir, exp_dir), "r") as f:
|
647 |
+
log = f.read()
|
648 |
+
print(log)
|
649 |
+
yield log
|
650 |
+
|
651 |
+
|
652 |
+
def change_sr2(sr2, if_f0_3, version19):
|
653 |
+
path_str = "" if version19 == "v1" else "_v2"
|
654 |
+
f0_str = "f0" if if_f0_3 else ""
|
655 |
+
if_pretrained_generator_exist = os.access("pretrained%s/%sG%s.pth" % (path_str, f0_str, sr2), os.F_OK)
|
656 |
+
if_pretrained_discriminator_exist = os.access("pretrained%s/%sD%s.pth" % (path_str, f0_str, sr2), os.F_OK)
|
657 |
+
if (if_pretrained_generator_exist == False):
|
658 |
+
print("pretrained%s/%sG%s.pth" % (path_str, f0_str, sr2), "not exist, will not use pretrained model")
|
659 |
+
if (if_pretrained_discriminator_exist == False):
|
660 |
+
print("pretrained%s/%sD%s.pth" % (path_str, f0_str, sr2), "not exist, will not use pretrained model")
|
661 |
+
return (
|
662 |
+
("pretrained%s/%sG%s.pth" % (path_str, f0_str, sr2)) if if_pretrained_generator_exist else "",
|
663 |
+
("pretrained%s/%sD%s.pth" % (path_str, f0_str, sr2)) if if_pretrained_discriminator_exist else "",
|
664 |
+
{"visible": True, "__type__": "update"}
|
665 |
+
)
|
666 |
+
|
667 |
+
def change_version19(sr2, if_f0_3, version19):
|
668 |
+
path_str = "" if version19 == "v1" else "_v2"
|
669 |
+
f0_str = "f0" if if_f0_3 else ""
|
670 |
+
if_pretrained_generator_exist = os.access("pretrained%s/%sG%s.pth" % (path_str, f0_str, sr2), os.F_OK)
|
671 |
+
if_pretrained_discriminator_exist = os.access("pretrained%s/%sD%s.pth" % (path_str, f0_str, sr2), os.F_OK)
|
672 |
+
if (if_pretrained_generator_exist == False):
|
673 |
+
print("pretrained%s/%sG%s.pth" % (path_str, f0_str, sr2), "not exist, will not use pretrained model")
|
674 |
+
if (if_pretrained_discriminator_exist == False):
|
675 |
+
print("pretrained%s/%sD%s.pth" % (path_str, f0_str, sr2), "not exist, will not use pretrained model")
|
676 |
+
return (
|
677 |
+
("pretrained%s/%sG%s.pth" % (path_str, f0_str, sr2)) if if_pretrained_generator_exist else "",
|
678 |
+
("pretrained%s/%sD%s.pth" % (path_str, f0_str, sr2)) if if_pretrained_discriminator_exist else "",
|
679 |
+
)
|
680 |
+
|
681 |
+
|
682 |
+
def change_f0(if_f0_3, sr2, version19): # f0method8,pretrained_G14,pretrained_D15
|
683 |
+
path_str = "" if version19 == "v1" else "_v2"
|
684 |
+
if_pretrained_generator_exist = os.access("pretrained%s/f0G%s.pth" % (path_str, sr2), os.F_OK)
|
685 |
+
if_pretrained_discriminator_exist = os.access("pretrained%s/f0D%s.pth" % (path_str, sr2), os.F_OK)
|
686 |
+
if (if_pretrained_generator_exist == False):
|
687 |
+
print("pretrained%s/f0G%s.pth" % (path_str, sr2), "not exist, will not use pretrained model")
|
688 |
+
if (if_pretrained_discriminator_exist == False):
|
689 |
+
print("pretrained%s/f0D%s.pth" % (path_str, sr2), "not exist, will not use pretrained model")
|
690 |
+
if if_f0_3:
|
691 |
+
return (
|
692 |
+
{"visible": True, "__type__": "update"},
|
693 |
+
"pretrained%s/f0G%s.pth" % (path_str, sr2) if if_pretrained_generator_exist else "",
|
694 |
+
"pretrained%s/f0D%s.pth" % (path_str, sr2) if if_pretrained_discriminator_exist else "",
|
695 |
+
)
|
696 |
+
return (
|
697 |
+
{"visible": False, "__type__": "update"},
|
698 |
+
("pretrained%s/G%s.pth" % (path_str, sr2)) if if_pretrained_generator_exist else "",
|
699 |
+
("pretrained%s/D%s.pth" % (path_str, sr2)) if if_pretrained_discriminator_exist else "",
|
700 |
+
)
|
701 |
+
|
702 |
+
|
703 |
+
global log_interval
|
704 |
+
|
705 |
+
|
706 |
+
def set_log_interval(exp_dir, batch_size12):
|
707 |
+
log_interval = 1
|
708 |
+
|
709 |
+
folder_path = os.path.join(exp_dir, "1_16k_wavs")
|
710 |
+
|
711 |
+
if os.path.exists(folder_path) and os.path.isdir(folder_path):
|
712 |
+
wav_files = [f for f in os.listdir(folder_path) if f.endswith(".wav")]
|
713 |
+
if wav_files:
|
714 |
+
sample_size = len(wav_files)
|
715 |
+
log_interval = math.ceil(sample_size / batch_size12)
|
716 |
+
if log_interval > 1:
|
717 |
+
log_interval += 1
|
718 |
+
return log_interval
|
719 |
+
|
720 |
+
# but3.click(click_train,[exp_dir1,sr2,if_f0_3,save_epoch10,total_epoch11,batch_size12,if_save_latest13,pretrained_G14,pretrained_D15,gpus16])
|
721 |
+
def click_train(
|
722 |
+
exp_dir1,
|
723 |
+
sr2,
|
724 |
+
if_f0_3,
|
725 |
+
spk_id5,
|
726 |
+
save_epoch10,
|
727 |
+
total_epoch11,
|
728 |
+
batch_size12,
|
729 |
+
if_save_latest13,
|
730 |
+
pretrained_G14,
|
731 |
+
pretrained_D15,
|
732 |
+
gpus16,
|
733 |
+
if_cache_gpu17,
|
734 |
+
if_save_every_weights18,
|
735 |
+
version19,
|
736 |
+
):
|
737 |
+
CSVutil('csvdb/stop.csv', 'w+', 'formanting', False)
|
738 |
+
# 生成filelist
|
739 |
+
exp_dir = "%s/logs/%s" % (now_dir, exp_dir1)
|
740 |
+
os.makedirs(exp_dir, exist_ok=True)
|
741 |
+
gt_wavs_dir = "%s/0_gt_wavs" % (exp_dir)
|
742 |
+
feature_dir = (
|
743 |
+
"%s/3_feature256" % (exp_dir)
|
744 |
+
if version19 == "v1"
|
745 |
+
else "%s/3_feature768" % (exp_dir)
|
746 |
+
)
|
747 |
+
|
748 |
+
log_interval = set_log_interval(exp_dir, batch_size12)
|
749 |
+
|
750 |
+
if if_f0_3:
|
751 |
+
f0_dir = "%s/2a_f0" % (exp_dir)
|
752 |
+
f0nsf_dir = "%s/2b-f0nsf" % (exp_dir)
|
753 |
+
names = (
|
754 |
+
set([name.split(".")[0] for name in os.listdir(gt_wavs_dir)])
|
755 |
+
& set([name.split(".")[0] for name in os.listdir(feature_dir)])
|
756 |
+
& set([name.split(".")[0] for name in os.listdir(f0_dir)])
|
757 |
+
& set([name.split(".")[0] for name in os.listdir(f0nsf_dir)])
|
758 |
+
)
|
759 |
+
else:
|
760 |
+
names = set([name.split(".")[0] for name in os.listdir(gt_wavs_dir)]) & set(
|
761 |
+
[name.split(".")[0] for name in os.listdir(feature_dir)]
|
762 |
+
)
|
763 |
+
opt = []
|
764 |
+
for name in names:
|
765 |
+
if if_f0_3:
|
766 |
+
opt.append(
|
767 |
+
"%s/%s.wav|%s/%s.npy|%s/%s.wav.npy|%s/%s.wav.npy|%s"
|
768 |
+
% (
|
769 |
+
gt_wavs_dir.replace("\\", "\\\\"),
|
770 |
+
name,
|
771 |
+
feature_dir.replace("\\", "\\\\"),
|
772 |
+
name,
|
773 |
+
f0_dir.replace("\\", "\\\\"),
|
774 |
+
name,
|
775 |
+
f0nsf_dir.replace("\\", "\\\\"),
|
776 |
+
name,
|
777 |
+
spk_id5,
|
778 |
+
)
|
779 |
+
)
|
780 |
+
else:
|
781 |
+
opt.append(
|
782 |
+
"%s/%s.wav|%s/%s.npy|%s"
|
783 |
+
% (
|
784 |
+
gt_wavs_dir.replace("\\", "\\\\"),
|
785 |
+
name,
|
786 |
+
feature_dir.replace("\\", "\\\\"),
|
787 |
+
name,
|
788 |
+
spk_id5,
|
789 |
+
)
|
790 |
+
)
|
791 |
+
fea_dim = 256 if version19 == "v1" else 768
|
792 |
+
if if_f0_3:
|
793 |
+
for _ in range(2):
|
794 |
+
opt.append(
|
795 |
+
"%s/logs/mute/0_gt_wavs/mute%s.wav|%s/logs/mute/3_feature%s/mute.npy|%s/logs/mute/2a_f0/mute.wav.npy|%s/logs/mute/2b-f0nsf/mute.wav.npy|%s"
|
796 |
+
% (now_dir, sr2, now_dir, fea_dim, now_dir, now_dir, spk_id5)
|
797 |
+
)
|
798 |
+
else:
|
799 |
+
for _ in range(2):
|
800 |
+
opt.append(
|
801 |
+
"%s/logs/mute/0_gt_wavs/mute%s.wav|%s/logs/mute/3_feature%s/mute.npy|%s"
|
802 |
+
% (now_dir, sr2, now_dir, fea_dim, spk_id5)
|
803 |
+
)
|
804 |
+
shuffle(opt)
|
805 |
+
with open("%s/filelist.txt" % exp_dir, "w") as f:
|
806 |
+
f.write("\n".join(opt))
|
807 |
+
print("write filelist done")
|
808 |
+
# 生成config#无需生成config
|
809 |
+
# cmd = python_cmd + " train_nsf_sim_cache_sid_load_pretrain.py -e mi-test -sr 40k -f0 1 -bs 4 -g 0 -te 10 -se 5 -pg pretrained/f0G40k.pth -pd pretrained/f0D40k.pth -l 1 -c 0"
|
810 |
+
print("use gpus:", gpus16)
|
811 |
+
if pretrained_G14 == "":
|
812 |
+
print("no pretrained Generator")
|
813 |
+
if pretrained_D15 == "":
|
814 |
+
print("no pretrained Discriminator")
|
815 |
+
if gpus16:
|
816 |
+
cmd = (
|
817 |
+
config.python_cmd
|
818 |
+
+ " train_nsf_sim_cache_sid_load_pretrain.py -e %s -sr %s -f0 %s -bs %s -g %s -te %s -se %s %s %s -l %s -c %s -sw %s -v %s -li %s"
|
819 |
+
% (
|
820 |
+
exp_dir1,
|
821 |
+
sr2,
|
822 |
+
1 if if_f0_3 else 0,
|
823 |
+
batch_size12,
|
824 |
+
gpus16,
|
825 |
+
total_epoch11,
|
826 |
+
save_epoch10,
|
827 |
+
("-pg %s" % pretrained_G14) if pretrained_G14 != "" else "",
|
828 |
+
("-pd %s" % pretrained_D15) if pretrained_D15 != "" else "",
|
829 |
+
1 if if_save_latest13 == True else 0,
|
830 |
+
1 if if_cache_gpu17 == True else 0,
|
831 |
+
1 if if_save_every_weights18 == True else 0,
|
832 |
+
version19,
|
833 |
+
log_interval,
|
834 |
+
)
|
835 |
+
)
|
836 |
+
else:
|
837 |
+
cmd = (
|
838 |
+
config.python_cmd
|
839 |
+
+ " train_nsf_sim_cache_sid_load_pretrain.py -e %s -sr %s -f0 %s -bs %s -te %s -se %s %s %s -l %s -c %s -sw %s -v %s -li %s"
|
840 |
+
% (
|
841 |
+
exp_dir1,
|
842 |
+
sr2,
|
843 |
+
1 if if_f0_3 else 0,
|
844 |
+
batch_size12,
|
845 |
+
total_epoch11,
|
846 |
+
save_epoch10,
|
847 |
+
("-pg %s" % pretrained_G14) if pretrained_G14 != "" else "\b",
|
848 |
+
("-pd %s" % pretrained_D15) if pretrained_D15 != "" else "\b",
|
849 |
+
1 if if_save_latest13 == True else 0,
|
850 |
+
1 if if_cache_gpu17 == True else 0,
|
851 |
+
1 if if_save_every_weights18 == True else 0,
|
852 |
+
version19,
|
853 |
+
log_interval,
|
854 |
+
)
|
855 |
+
)
|
856 |
+
print(cmd)
|
857 |
+
p = Popen(cmd, shell=True, cwd=now_dir)
|
858 |
+
global PID
|
859 |
+
PID = p.pid
|
860 |
+
p.wait()
|
861 |
+
return ("训练结束, 您可查看控制台训练日志或实验文件夹下的train.log", {"visible": False, "__type__": "update"}, {"visible": True, "__type__": "update"})
|
862 |
+
|
863 |
+
|
864 |
+
# but4.click(train_index, [exp_dir1], info3)
|
865 |
+
def train_index(exp_dir1, version19):
|
866 |
+
exp_dir = "%s/logs/%s" % (now_dir, exp_dir1)
|
867 |
+
os.makedirs(exp_dir, exist_ok=True)
|
868 |
+
feature_dir = (
|
869 |
+
"%s/3_feature256" % (exp_dir)
|
870 |
+
if version19 == "v1"
|
871 |
+
else "%s/3_feature768" % (exp_dir)
|
872 |
+
)
|
873 |
+
if os.path.exists(feature_dir) == False:
|
874 |
+
return "请先进行特征提取!"
|
875 |
+
listdir_res = list(os.listdir(feature_dir))
|
876 |
+
if len(listdir_res) == 0:
|
877 |
+
return "请先进行特征提取!"
|
878 |
+
npys = []
|
879 |
+
for name in sorted(listdir_res):
|
880 |
+
phone = np.load("%s/%s" % (feature_dir, name))
|
881 |
+
npys.append(phone)
|
882 |
+
big_npy = np.concatenate(npys, 0)
|
883 |
+
big_npy_idx = np.arange(big_npy.shape[0])
|
884 |
+
np.random.shuffle(big_npy_idx)
|
885 |
+
big_npy = big_npy[big_npy_idx]
|
886 |
+
np.save("%s/total_fea.npy" % exp_dir, big_npy)
|
887 |
+
# n_ivf = big_npy.shape[0] // 39
|
888 |
+
n_ivf = min(int(16 * np.sqrt(big_npy.shape[0])), big_npy.shape[0] // 39)
|
889 |
+
infos = []
|
890 |
+
infos.append("%s,%s" % (big_npy.shape, n_ivf))
|
891 |
+
yield "\n".join(infos)
|
892 |
+
index = faiss.index_factory(256 if version19 == "v1" else 768, "IVF%s,Flat" % n_ivf)
|
893 |
+
# index = faiss.index_factory(256if version19=="v1"else 768, "IVF%s,PQ128x4fs,RFlat"%n_ivf)
|
894 |
+
infos.append("training")
|
895 |
+
yield "\n".join(infos)
|
896 |
+
index_ivf = faiss.extract_index_ivf(index) #
|
897 |
+
index_ivf.nprobe = 1
|
898 |
+
index.train(big_npy)
|
899 |
+
faiss.write_index(
|
900 |
+
index,
|
901 |
+
"%s/trained_IVF%s_Flat_nprobe_%s_%s_%s.index"
|
902 |
+
% (exp_dir, n_ivf, index_ivf.nprobe, exp_dir1, version19),
|
903 |
+
)
|
904 |
+
# faiss.write_index(index, '%s/trained_IVF%s_Flat_FastScan_%s.index'%(exp_dir,n_ivf,version19))
|
905 |
+
infos.append("adding")
|
906 |
+
yield "\n".join(infos)
|
907 |
+
batch_size_add = 8192
|
908 |
+
for i in range(0, big_npy.shape[0], batch_size_add):
|
909 |
+
index.add(big_npy[i : i + batch_size_add])
|
910 |
+
faiss.write_index(
|
911 |
+
index,
|
912 |
+
"%s/added_IVF%s_Flat_nprobe_%s_%s_%s.index"
|
913 |
+
% (exp_dir, n_ivf, index_ivf.nprobe, exp_dir1, version19),
|
914 |
+
)
|
915 |
+
infos.append(
|
916 |
+
"成功构建索引,added_IVF%s_Flat_nprobe_%s_%s_%s.index"
|
917 |
+
% (n_ivf, index_ivf.nprobe, exp_dir1, version19)
|
918 |
+
)
|
919 |
+
# faiss.write_index(index, '%s/added_IVF%s_Flat_FastScan_%s.index'%(exp_dir,n_ivf,version19))
|
920 |
+
# infos.append("成功构建索引,added_IVF%s_Flat_FastScan_%s.index"%(n_ivf,version19))
|
921 |
+
yield "\n".join(infos)
|
922 |
+
|
923 |
+
|
924 |
+
# but5.click(train1key, [exp_dir1, sr2, if_f0_3, trainset_dir4, spk_id5, gpus6, np7, f0method8, save_epoch10, total_epoch11, batch_size12, if_save_latest13, pretrained_G14, pretrained_D15, gpus16, if_cache_gpu17], info3)
|
925 |
+
def train1key(
|
926 |
+
exp_dir1,
|
927 |
+
sr2,
|
928 |
+
if_f0_3,
|
929 |
+
trainset_dir4,
|
930 |
+
spk_id5,
|
931 |
+
np7,
|
932 |
+
f0method8,
|
933 |
+
save_epoch10,
|
934 |
+
total_epoch11,
|
935 |
+
batch_size12,
|
936 |
+
if_save_latest13,
|
937 |
+
pretrained_G14,
|
938 |
+
pretrained_D15,
|
939 |
+
gpus16,
|
940 |
+
if_cache_gpu17,
|
941 |
+
if_save_every_weights18,
|
942 |
+
version19,
|
943 |
+
echl
|
944 |
+
):
|
945 |
+
infos = []
|
946 |
+
|
947 |
+
def get_info_str(strr):
|
948 |
+
infos.append(strr)
|
949 |
+
return "\n".join(infos)
|
950 |
+
|
951 |
+
model_log_dir = "%s/logs/%s" % (now_dir, exp_dir1)
|
952 |
+
preprocess_log_path = "%s/preprocess.log" % model_log_dir
|
953 |
+
extract_f0_feature_log_path = "%s/extract_f0_feature.log" % model_log_dir
|
954 |
+
gt_wavs_dir = "%s/0_gt_wavs" % model_log_dir
|
955 |
+
feature_dir = (
|
956 |
+
"%s/3_feature256" % model_log_dir
|
957 |
+
if version19 == "v1"
|
958 |
+
else "%s/3_feature768" % model_log_dir
|
959 |
+
)
|
960 |
+
|
961 |
+
os.makedirs(model_log_dir, exist_ok=True)
|
962 |
+
#########step1:处理数据
|
963 |
+
open(preprocess_log_path, "w").close()
|
964 |
+
cmd = (
|
965 |
+
config.python_cmd
|
966 |
+
+ " trainset_preprocess_pipeline_print.py %s %s %s %s "
|
967 |
+
% (trainset_dir4, sr_dict[sr2], np7, model_log_dir)
|
968 |
+
+ str(config.noparallel)
|
969 |
+
)
|
970 |
+
yield get_info_str(i18n("step1:正在处理数据"))
|
971 |
+
yield get_info_str(cmd)
|
972 |
+
p = Popen(cmd, shell=True)
|
973 |
+
p.wait()
|
974 |
+
with open(preprocess_log_path, "r") as f:
|
975 |
+
print(f.read())
|
976 |
+
#########step2a:提取音高
|
977 |
+
open(extract_f0_feature_log_path, "w")
|
978 |
+
if if_f0_3:
|
979 |
+
yield get_info_str("step2a:正在提取音高")
|
980 |
+
cmd = config.python_cmd + " extract_f0_print.py %s %s %s %s" % (
|
981 |
+
model_log_dir,
|
982 |
+
np7,
|
983 |
+
f0method8,
|
984 |
+
echl
|
985 |
+
)
|
986 |
+
yield get_info_str(cmd)
|
987 |
+
p = Popen(cmd, shell=True, cwd=now_dir)
|
988 |
+
p.wait()
|
989 |
+
with open(extract_f0_feature_log_path, "r") as f:
|
990 |
+
print(f.read())
|
991 |
+
else:
|
992 |
+
yield get_info_str(i18n("step2a:无需提取音高"))
|
993 |
+
#######step2b:提取特征
|
994 |
+
yield get_info_str(i18n("step2b:正在提取特征"))
|
995 |
+
gpus = gpus16.split("-")
|
996 |
+
leng = len(gpus)
|
997 |
+
ps = []
|
998 |
+
for idx, n_g in enumerate(gpus):
|
999 |
+
cmd = config.python_cmd + " extract_feature_print.py %s %s %s %s %s %s" % (
|
1000 |
+
config.device,
|
1001 |
+
leng,
|
1002 |
+
idx,
|
1003 |
+
n_g,
|
1004 |
+
model_log_dir,
|
1005 |
+
version19,
|
1006 |
+
)
|
1007 |
+
yield get_info_str(cmd)
|
1008 |
+
p = Popen(
|
1009 |
+
cmd, shell=True, cwd=now_dir
|
1010 |
+
) # , shell=True, stdin=PIPE, stdout=PIPE, stderr=PIPE, cwd=now_dir
|
1011 |
+
ps.append(p)
|
1012 |
+
for p in ps:
|
1013 |
+
p.wait()
|
1014 |
+
with open(extract_f0_feature_log_path, "r") as f:
|
1015 |
+
print(f.read())
|
1016 |
+
#######step3a:训练模型
|
1017 |
+
yield get_info_str(i18n("step3a:正在训练模型"))
|
1018 |
+
# 生成filelist
|
1019 |
+
if if_f0_3:
|
1020 |
+
f0_dir = "%s/2a_f0" % model_log_dir
|
1021 |
+
f0nsf_dir = "%s/2b-f0nsf" % model_log_dir
|
1022 |
+
names = (
|
1023 |
+
set([name.split(".")[0] for name in os.listdir(gt_wavs_dir)])
|
1024 |
+
& set([name.split(".")[0] for name in os.listdir(feature_dir)])
|
1025 |
+
& set([name.split(".")[0] for name in os.listdir(f0_dir)])
|
1026 |
+
& set([name.split(".")[0] for name in os.listdir(f0nsf_dir)])
|
1027 |
+
)
|
1028 |
+
else:
|
1029 |
+
names = set([name.split(".")[0] for name in os.listdir(gt_wavs_dir)]) & set(
|
1030 |
+
[name.split(".")[0] for name in os.listdir(feature_dir)]
|
1031 |
+
)
|
1032 |
+
opt = []
|
1033 |
+
for name in names:
|
1034 |
+
if if_f0_3:
|
1035 |
+
opt.append(
|
1036 |
+
"%s/%s.wav|%s/%s.npy|%s/%s.wav.npy|%s/%s.wav.npy|%s"
|
1037 |
+
% (
|
1038 |
+
gt_wavs_dir.replace("\\", "\\\\"),
|
1039 |
+
name,
|
1040 |
+
feature_dir.replace("\\", "\\\\"),
|
1041 |
+
name,
|
1042 |
+
f0_dir.replace("\\", "\\\\"),
|
1043 |
+
name,
|
1044 |
+
f0nsf_dir.replace("\\", "\\\\"),
|
1045 |
+
name,
|
1046 |
+
spk_id5,
|
1047 |
+
)
|
1048 |
+
)
|
1049 |
+
else:
|
1050 |
+
opt.append(
|
1051 |
+
"%s/%s.wav|%s/%s.npy|%s"
|
1052 |
+
% (
|
1053 |
+
gt_wavs_dir.replace("\\", "\\\\"),
|
1054 |
+
name,
|
1055 |
+
feature_dir.replace("\\", "\\\\"),
|
1056 |
+
name,
|
1057 |
+
spk_id5,
|
1058 |
+
)
|
1059 |
+
)
|
1060 |
+
fea_dim = 256 if version19 == "v1" else 768
|
1061 |
+
if if_f0_3:
|
1062 |
+
for _ in range(2):
|
1063 |
+
opt.append(
|
1064 |
+
"%s/logs/mute/0_gt_wavs/mute%s.wav|%s/logs/mute/3_feature%s/mute.npy|%s/logs/mute/2a_f0/mute.wav.npy|%s/logs/mute/2b-f0nsf/mute.wav.npy|%s"
|
1065 |
+
% (now_dir, sr2, now_dir, fea_dim, now_dir, now_dir, spk_id5)
|
1066 |
+
)
|
1067 |
+
else:
|
1068 |
+
for _ in range(2):
|
1069 |
+
opt.append(
|
1070 |
+
"%s/logs/mute/0_gt_wavs/mute%s.wav|%s/logs/mute/3_feature%s/mute.npy|%s"
|
1071 |
+
% (now_dir, sr2, now_dir, fea_dim, spk_id5)
|
1072 |
+
)
|
1073 |
+
shuffle(opt)
|
1074 |
+
with open("%s/filelist.txt" % model_log_dir, "w") as f:
|
1075 |
+
f.write("\n".join(opt))
|
1076 |
+
yield get_info_str("write filelist done")
|
1077 |
+
if gpus16:
|
1078 |
+
cmd = (
|
1079 |
+
config.python_cmd
|
1080 |
+
+" train_nsf_sim_cache_sid_load_pretrain.py -e %s -sr %s -f0 %s -bs %s -g %s -te %s -se %s %s %s -l %s -c %s -sw %s -v %s"
|
1081 |
+
% (
|
1082 |
+
exp_dir1,
|
1083 |
+
sr2,
|
1084 |
+
1 if if_f0_3 else 0,
|
1085 |
+
batch_size12,
|
1086 |
+
gpus16,
|
1087 |
+
total_epoch11,
|
1088 |
+
save_epoch10,
|
1089 |
+
("-pg %s" % pretrained_G14) if pretrained_G14 != "" else "",
|
1090 |
+
("-pd %s" % pretrained_D15) if pretrained_D15 != "" else "",
|
1091 |
+
1 if if_save_latest13 == True else 0,
|
1092 |
+
1 if if_cache_gpu17 == True else 0,
|
1093 |
+
1 if if_save_every_weights18 == True else 0,
|
1094 |
+
version19,
|
1095 |
+
)
|
1096 |
+
)
|
1097 |
+
else:
|
1098 |
+
cmd = (
|
1099 |
+
config.python_cmd
|
1100 |
+
+ " train_nsf_sim_cache_sid_load_pretrain.py -e %s -sr %s -f0 %s -bs %s -te %s -se %s %s %s -l %s -c %s -sw %s -v %s"
|
1101 |
+
% (
|
1102 |
+
exp_dir1,
|
1103 |
+
sr2,
|
1104 |
+
1 if if_f0_3 else 0,
|
1105 |
+
batch_size12,
|
1106 |
+
total_epoch11,
|
1107 |
+
save_epoch10,
|
1108 |
+
("-pg %s" % pretrained_G14) if pretrained_G14 != "" else "",
|
1109 |
+
("-pd %s" % pretrained_D15) if pretrained_D15 != "" else "",
|
1110 |
+
1 if if_save_latest13 == True else 0,
|
1111 |
+
1 if if_cache_gpu17 == True else 0,
|
1112 |
+
1 if if_save_every_weights18 == True else 0,
|
1113 |
+
version19,
|
1114 |
+
)
|
1115 |
+
)
|
1116 |
+
yield get_info_str(cmd)
|
1117 |
+
p = Popen(cmd, shell=True, cwd=now_dir)
|
1118 |
+
p.wait()
|
1119 |
+
yield get_info_str(i18n("训练结束, 您可查看控制台训练日志或实验文件夹下的train.log"))
|
1120 |
+
#######step3b:训练索引
|
1121 |
+
npys = []
|
1122 |
+
listdir_res = list(os.listdir(feature_dir))
|
1123 |
+
for name in sorted(listdir_res):
|
1124 |
+
phone = np.load("%s/%s" % (feature_dir, name))
|
1125 |
+
npys.append(phone)
|
1126 |
+
big_npy = np.concatenate(npys, 0)
|
1127 |
+
|
1128 |
+
big_npy_idx = np.arange(big_npy.shape[0])
|
1129 |
+
np.random.shuffle(big_npy_idx)
|
1130 |
+
big_npy = big_npy[big_npy_idx]
|
1131 |
+
np.save("%s/total_fea.npy" % model_log_dir, big_npy)
|
1132 |
+
|
1133 |
+
# n_ivf = big_npy.shape[0] // 39
|
1134 |
+
n_ivf = min(int(16 * np.sqrt(big_npy.shape[0])), big_npy.shape[0] // 39)
|
1135 |
+
yield get_info_str("%s,%s" % (big_npy.shape, n_ivf))
|
1136 |
+
index = faiss.index_factory(256 if version19 == "v1" else 768, "IVF%s,Flat" % n_ivf)
|
1137 |
+
yield get_info_str("training index")
|
1138 |
+
index_ivf = faiss.extract_index_ivf(index) #
|
1139 |
+
index_ivf.nprobe = 1
|
1140 |
+
index.train(big_npy)
|
1141 |
+
faiss.write_index(
|
1142 |
+
index,
|
1143 |
+
"%s/trained_IVF%s_Flat_nprobe_%s_%s_%s.index"
|
1144 |
+
% (model_log_dir, n_ivf, index_ivf.nprobe, exp_dir1, version19),
|
1145 |
+
)
|
1146 |
+
yield get_info_str("adding index")
|
1147 |
+
batch_size_add = 8192
|
1148 |
+
for i in range(0, big_npy.shape[0], batch_size_add):
|
1149 |
+
index.add(big_npy[i : i + batch_size_add])
|
1150 |
+
faiss.write_index(
|
1151 |
+
index,
|
1152 |
+
"%s/added_IVF%s_Flat_nprobe_%s_%s_%s.index"
|
1153 |
+
% (model_log_dir, n_ivf, index_ivf.nprobe, exp_dir1, version19),
|
1154 |
+
)
|
1155 |
+
yield get_info_str(
|
1156 |
+
"成功构建索引, added_IVF%s_Flat_nprobe_%s_%s_%s.index"
|
1157 |
+
% (n_ivf, index_ivf.nprobe, exp_dir1, version19)
|
1158 |
+
)
|
1159 |
+
yield get_info_str(i18n("全流程结束!"))
|
1160 |
+
|
1161 |
+
|
1162 |
+
def whethercrepeornah(radio):
|
1163 |
+
mango = True if radio == 'mangio-crepe' or radio == 'mangio-crepe-tiny' else False
|
1164 |
+
return ({"visible": mango, "__type__": "update"})
|
1165 |
+
|
1166 |
+
# ckpt_path2.change(change_info_,[ckpt_path2],[sr__,if_f0__])
|
1167 |
+
def change_info_(ckpt_path):
|
1168 |
+
if (
|
1169 |
+
os.path.exists(ckpt_path.replace(os.path.basename(ckpt_path), "train.log"))
|
1170 |
+
== False
|
1171 |
+
):
|
1172 |
+
return {"__type__": "update"}, {"__type__": "update"}, {"__type__": "update"}
|
1173 |
+
try:
|
1174 |
+
with open(
|
1175 |
+
ckpt_path.replace(os.path.basename(ckpt_path), "train.log"), "r"
|
1176 |
+
) as f:
|
1177 |
+
info = eval(f.read().strip("\n").split("\n")[0].split("\t")[-1])
|
1178 |
+
sr, f0 = info["sample_rate"], info["if_f0"]
|
1179 |
+
version = "v2" if ("version" in info and info["version"] == "v2") else "v1"
|
1180 |
+
return sr, str(f0), version
|
1181 |
+
except:
|
1182 |
+
traceback.print_exc()
|
1183 |
+
return {"__type__": "update"}, {"__type__": "update"}, {"__type__": "update"}
|
1184 |
+
|
1185 |
+
|
1186 |
+
from lib.infer_pack.models_onnx import SynthesizerTrnMsNSFsidM
|
1187 |
+
|
1188 |
+
|
1189 |
+
def export_onnx(ModelPath, ExportedPath, MoeVS=True):
|
1190 |
+
cpt = torch.load(ModelPath, map_location="cpu")
|
1191 |
+
cpt["config"][-3] = cpt["weight"]["emb_g.weight"].shape[0] # n_spk
|
1192 |
+
hidden_channels = 256 if cpt.get("version","v1")=="v1"else 768#cpt["config"][-2] # hidden_channels,为768Vec做准备
|
1193 |
+
|
1194 |
+
test_phone = torch.rand(1, 200, hidden_channels) # hidden unit
|
1195 |
+
test_phone_lengths = torch.tensor([200]).long() # hidden unit 长度(貌似没啥用)
|
1196 |
+
test_pitch = torch.randint(size=(1, 200), low=5, high=255) # 基频(单位赫兹)
|
1197 |
+
test_pitchf = torch.rand(1, 200) # nsf基频
|
1198 |
+
test_ds = torch.LongTensor([0]) # 说话人ID
|
1199 |
+
test_rnd = torch.rand(1, 192, 200) # 噪声(加入随机因子)
|
1200 |
+
|
1201 |
+
device = "cpu" # 导出时设备(不影响使用模型)
|
1202 |
+
|
1203 |
+
|
1204 |
+
net_g = SynthesizerTrnMsNSFsidM(
|
1205 |
+
*cpt["config"], is_half=False,version=cpt.get("version","v1")
|
1206 |
+
) # fp32导出(C++要支持fp16必须手动将内存重新排列所以暂时不用fp16)
|
1207 |
+
net_g.load_state_dict(cpt["weight"], strict=False)
|
1208 |
+
input_names = ["phone", "phone_lengths", "pitch", "pitchf", "ds", "rnd"]
|
1209 |
+
output_names = [
|
1210 |
+
"audio",
|
1211 |
+
]
|
1212 |
+
# net_g.construct_spkmixmap(n_speaker) 多角色混合轨道导出
|
1213 |
+
torch.onnx.export(
|
1214 |
+
net_g,
|
1215 |
+
(
|
1216 |
+
test_phone.to(device),
|
1217 |
+
test_phone_lengths.to(device),
|
1218 |
+
test_pitch.to(device),
|
1219 |
+
test_pitchf.to(device),
|
1220 |
+
test_ds.to(device),
|
1221 |
+
test_rnd.to(device),
|
1222 |
+
),
|
1223 |
+
ExportedPath,
|
1224 |
+
dynamic_axes={
|
1225 |
+
"phone": [1],
|
1226 |
+
"pitch": [1],
|
1227 |
+
"pitchf": [1],
|
1228 |
+
"rnd": [2],
|
1229 |
+
},
|
1230 |
+
do_constant_folding=False,
|
1231 |
+
opset_version=16,
|
1232 |
+
verbose=False,
|
1233 |
+
input_names=input_names,
|
1234 |
+
output_names=output_names,
|
1235 |
+
)
|
1236 |
+
return "Finished"
|
1237 |
+
|
1238 |
+
#region RVC WebUI App
|
1239 |
+
|
1240 |
+
def get_presets():
|
1241 |
+
data = None
|
1242 |
+
with open('../inference-presets.json', 'r') as file:
|
1243 |
+
data = json.load(file)
|
1244 |
+
preset_names = []
|
1245 |
+
for preset in data['presets']:
|
1246 |
+
preset_names.append(preset['name'])
|
1247 |
+
|
1248 |
+
return preset_names
|
1249 |
+
|
1250 |
+
def change_choices2():
|
1251 |
+
audio_files=[]
|
1252 |
+
for filename in os.listdir("./audios"):
|
1253 |
+
if filename.endswith(('.wav','.mp3','.ogg','.flac','.m4a','.aac','.mp4')):
|
1254 |
+
audio_files.append(os.path.join('./audios',filename).replace('\\', '/'))
|
1255 |
+
return {"choices": sorted(audio_files), "__type__": "update"}, {"__type__": "update"}
|
1256 |
+
|
1257 |
+
audio_files=[]
|
1258 |
+
for filename in os.listdir("./audios"):
|
1259 |
+
if filename.endswith(('.wav','.mp3','.ogg','.flac','.m4a','.aac','.mp4')):
|
1260 |
+
audio_files.append(os.path.join('./audios',filename).replace('\\', '/'))
|
1261 |
+
|
1262 |
+
def get_index():
|
1263 |
+
if check_for_name() != '':
|
1264 |
+
chosen_model=sorted(names)[0].split(".")[0]
|
1265 |
+
logs_path="./logs/"+chosen_model
|
1266 |
+
if os.path.exists(logs_path):
|
1267 |
+
for file in os.listdir(logs_path):
|
1268 |
+
if file.endswith(".index"):
|
1269 |
+
return os.path.join(logs_path, file)
|
1270 |
+
return ''
|
1271 |
+
else:
|
1272 |
+
return ''
|
1273 |
+
|
1274 |
+
def get_indexes():
|
1275 |
+
indexes_list=[]
|
1276 |
+
for dirpath, dirnames, filenames in os.walk("./logs/"):
|
1277 |
+
for filename in filenames:
|
1278 |
+
if filename.endswith(".index"):
|
1279 |
+
indexes_list.append(os.path.join(dirpath,filename))
|
1280 |
+
if len(indexes_list) > 0:
|
1281 |
+
return indexes_list
|
1282 |
+
else:
|
1283 |
+
return ''
|
1284 |
+
|
1285 |
+
def get_name():
|
1286 |
+
if len(audio_files) > 0:
|
1287 |
+
return sorted(audio_files)[0]
|
1288 |
+
else:
|
1289 |
+
return ''
|
1290 |
+
|
1291 |
+
def save_to_wav(record_button):
|
1292 |
+
if record_button is None:
|
1293 |
+
pass
|
1294 |
+
else:
|
1295 |
+
path_to_file=record_button
|
1296 |
+
new_name = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")+'.wav'
|
1297 |
+
new_path='./audios/'+new_name
|
1298 |
+
shutil.move(path_to_file,new_path)
|
1299 |
+
return new_path
|
1300 |
+
|
1301 |
+
def save_to_wav2(dropbox):
|
1302 |
+
file_path=dropbox.name
|
1303 |
+
shutil.move(file_path,'./audios')
|
1304 |
+
return os.path.join('./audios',os.path.basename(file_path))
|
1305 |
+
|
1306 |
+
def match_index(sid0):
|
1307 |
+
folder=sid0.split(".")[0]
|
1308 |
+
parent_dir="./logs/"+folder
|
1309 |
+
if os.path.exists(parent_dir):
|
1310 |
+
for filename in os.listdir(parent_dir):
|
1311 |
+
if filename.endswith(".index"):
|
1312 |
+
index_path=os.path.join(parent_dir,filename)
|
1313 |
+
return index_path
|
1314 |
+
else:
|
1315 |
+
return ''
|
1316 |
+
|
1317 |
+
def check_for_name():
|
1318 |
+
if len(names) > 0:
|
1319 |
+
return sorted(names)[0]
|
1320 |
+
else:
|
1321 |
+
return ''
|
1322 |
+
|
1323 |
+
def download_from_url(url, model):
|
1324 |
+
if url == '':
|
1325 |
+
return "URL cannot be left empty."
|
1326 |
+
if model =='':
|
1327 |
+
return "You need to name your model. For example: My-Model"
|
1328 |
+
url = url.strip()
|
1329 |
+
zip_dirs = ["zips", "unzips"]
|
1330 |
+
for directory in zip_dirs:
|
1331 |
+
if os.path.exists(directory):
|
1332 |
+
shutil.rmtree(directory)
|
1333 |
+
os.makedirs("zips", exist_ok=True)
|
1334 |
+
os.makedirs("unzips", exist_ok=True)
|
1335 |
+
zipfile = model + '.zip'
|
1336 |
+
zipfile_path = './zips/' + zipfile
|
1337 |
+
try:
|
1338 |
+
if "drive.google.com" in url:
|
1339 |
+
subprocess.run(["gdown", url, "--fuzzy", "-O", zipfile_path])
|
1340 |
+
elif "mega.nz" in url:
|
1341 |
+
m = Mega()
|
1342 |
+
m.download_url(url, './zips')
|
1343 |
+
else:
|
1344 |
+
subprocess.run(["wget", url, "-O", zipfile_path])
|
1345 |
+
for filename in os.listdir("./zips"):
|
1346 |
+
if filename.endswith(".zip"):
|
1347 |
+
zipfile_path = os.path.join("./zips/",filename)
|
1348 |
+
shutil.unpack_archive(zipfile_path, "./unzips", 'zip')
|
1349 |
+
else:
|
1350 |
+
return "No zipfile found."
|
1351 |
+
for root, dirs, files in os.walk('./unzips'):
|
1352 |
+
for file in files:
|
1353 |
+
file_path = os.path.join(root, file)
|
1354 |
+
if file.endswith(".index"):
|
1355 |
+
os.mkdir(f'./logs/{model}')
|
1356 |
+
shutil.copy2(file_path,f'./logs/{model}')
|
1357 |
+
elif "G_" not in file and "D_" not in file and file.endswith(".pth"):
|
1358 |
+
shutil.copy(file_path,f'./weights/{model}.pth')
|
1359 |
+
shutil.rmtree("zips")
|
1360 |
+
shutil.rmtree("unzips")
|
1361 |
+
return "Modello scaricato con successo, puoi tornare all'inferenza."
|
1362 |
+
except:
|
1363 |
+
return "Errore, il modello non si è scaricato correttamente."
|
1364 |
+
def success_message(face):
|
1365 |
+
return f'{face.name} has been uploaded.', 'None'
|
1366 |
+
def mouth(size, face, voice, faces):
|
1367 |
+
if size == 'Half':
|
1368 |
+
size = 2
|
1369 |
+
else:
|
1370 |
+
size = 1
|
1371 |
+
if faces == 'None':
|
1372 |
+
character = face.name
|
1373 |
+
else:
|
1374 |
+
if faces == 'Ben Shapiro':
|
1375 |
+
character = '/content/wav2lip-HD/inputs/ben-shapiro-10.mp4'
|
1376 |
+
elif faces == 'Andrew Tate':
|
1377 |
+
character = '/content/wav2lip-HD/inputs/tate-7.mp4'
|
1378 |
+
command = "python inference.py " \
|
1379 |
+
"--checkpoint_path checkpoints/wav2lip.pth " \
|
1380 |
+
f"--face {character} " \
|
1381 |
+
f"--audio {voice} " \
|
1382 |
+
"--pads 0 20 0 0 " \
|
1383 |
+
"--outfile /content/wav2lip-HD/outputs/result.mp4 " \
|
1384 |
+
"--fps 24 " \
|
1385 |
+
f"--resize_factor {size}"
|
1386 |
+
process = subprocess.Popen(command, shell=True, cwd='/content/wav2lip-HD/Wav2Lip-master')
|
1387 |
+
stdout, stderr = process.communicate()
|
1388 |
+
return '/content/wav2lip-HD/outputs/result.mp4', 'Animation completed.'
|
1389 |
+
eleven_voices = ['Adam','Antoni','Josh','Arnold','Sam','Bella','Rachel','Domi','Elli']
|
1390 |
+
eleven_voices_ids=['pNInz6obpgDQGcFmaJgB','ErXwobaYiN019PkySvjV','TxGEqnHWrfWFTfGW9XjX','VR6AewLTigWG4xSOukaG','yoZ06aMxZJJ28mfd3POQ','EXAVITQu4vr4xnSDxMaL','21m00Tcm4TlvDq8ikWAM','AZnzlk1XvdvUeBnXmlld','MF3mGyEYCl7XYWbV9V6O']
|
1391 |
+
chosen_voice = dict(zip(eleven_voices, eleven_voices_ids))
|
1392 |
+
|
1393 |
+
def stoptraining(mim):
|
1394 |
+
if int(mim) == 1:
|
1395 |
+
try:
|
1396 |
+
CSVutil('csvdb/stop.csv', 'w+', 'stop', 'True')
|
1397 |
+
os.kill(PID, signal.SIGTERM)
|
1398 |
+
except Exception as e:
|
1399 |
+
print(f"Couldn't click due to {e}")
|
1400 |
+
return (
|
1401 |
+
{"visible": False, "__type__": "update"},
|
1402 |
+
{"visible": True, "__type__": "update"},
|
1403 |
+
)
|
1404 |
+
|
1405 |
+
|
1406 |
+
def elevenTTS(xiapi, text, id, lang):
|
1407 |
+
if xiapi!= '' and id !='':
|
1408 |
+
choice = chosen_voice[id]
|
1409 |
+
CHUNK_SIZE = 1024
|
1410 |
+
url = f"https://api.elevenlabs.io/v1/text-to-speech/{choice}"
|
1411 |
+
headers = {
|
1412 |
+
"Accept": "audio/mpeg",
|
1413 |
+
"Content-Type": "application/json",
|
1414 |
+
"xi-api-key": xiapi
|
1415 |
+
}
|
1416 |
+
if lang == 'en':
|
1417 |
+
data = {
|
1418 |
+
"text": text,
|
1419 |
+
"model_id": "eleven_monolingual_v1",
|
1420 |
+
"voice_settings": {
|
1421 |
+
"stability": 0.5,
|
1422 |
+
"similarity_boost": 0.5
|
1423 |
+
}
|
1424 |
+
}
|
1425 |
+
else:
|
1426 |
+
data = {
|
1427 |
+
"text": text,
|
1428 |
+
"model_id": "eleven_multilingual_v1",
|
1429 |
+
"voice_settings": {
|
1430 |
+
"stability": 0.5,
|
1431 |
+
"similarity_boost": 0.5
|
1432 |
+
}
|
1433 |
+
}
|
1434 |
+
|
1435 |
+
response = requests.post(url, json=data, headers=headers)
|
1436 |
+
with open('./temp_eleven.mp3', 'wb') as f:
|
1437 |
+
for chunk in response.iter_content(chunk_size=CHUNK_SIZE):
|
1438 |
+
if chunk:
|
1439 |
+
f.write(chunk)
|
1440 |
+
aud_path = save_to_wav('./temp_eleven.mp3')
|
1441 |
+
return aud_path, aud_path
|
1442 |
+
else:
|
1443 |
+
tts = gTTS(text, lang=lang)
|
1444 |
+
tts.save('./temp_gTTS.mp3')
|
1445 |
+
aud_path = save_to_wav('./temp_gTTS.mp3')
|
1446 |
+
return aud_path, aud_path
|
1447 |
+
|
1448 |
+
def upload_to_dataset(files, dir):
|
1449 |
+
if dir == '':
|
1450 |
+
dir = './dataset'
|
1451 |
+
if not os.path.exists(dir):
|
1452 |
+
os.makedirs(dir)
|
1453 |
+
count = 0
|
1454 |
+
for file in files:
|
1455 |
+
path=file.name
|
1456 |
+
shutil.copy2(path,dir)
|
1457 |
+
count += 1
|
1458 |
+
return f' {count} files uploaded to {dir}.'
|
1459 |
+
|
1460 |
+
def zip_downloader(model):
|
1461 |
+
if not os.path.exists(f'./weights/{model}.pth'):
|
1462 |
+
return {"__type__": "update"}, f'Make sure the Voice Name is correct. I could not find {model}.pth'
|
1463 |
+
index_found = False
|
1464 |
+
for file in os.listdir(f'./logs/{model}'):
|
1465 |
+
if file.endswith('.index') and 'added' in file:
|
1466 |
+
log_file = file
|
1467 |
+
index_found = True
|
1468 |
+
if index_found:
|
1469 |
+
return [f'./weights/{model}.pth', f'./logs/{model}/{log_file}'], "Done"
|
1470 |
+
else:
|
1471 |
+
return f'./weights/{model}.pth', "Could not find Index file."
|
1472 |
+
|
1473 |
+
with gr.Blocks(theme=gr.themes.Base (), title='Mangio-RVC-Web 💻') as app:
|
1474 |
+
with gr.Tabs():
|
1475 |
+
with gr.TabItem("Inferenza"):
|
1476 |
+
gr.HTML("<h1> Ilaria RVC 2.0 💖 </h1>")
|
1477 |
+
gr.HTML("<h10> Lista dei modelli a cura di Ilaria: https://rentry.org/modelli-italiani </h10>")
|
1478 |
+
gr.HTML("<h4> Sostieni Ilaria su Ko-Fi donando qualsiasi somma: https://ko-fi.com/ilariaowo </h4>")
|
1479 |
+
|
1480 |
+
# Inference Preset Row
|
1481 |
+
# with gr.Row():
|
1482 |
+
# mangio_preset = gr.Dropdown(label="Inference Preset", choices=sorted(get_presets()))
|
1483 |
+
# mangio_preset_name_save = gr.Textbox(
|
1484 |
+
# label="Your preset name"
|
1485 |
+
# )
|
1486 |
+
# mangio_preset_save_btn = gr.Button('Save Preset', variant="primary")
|
1487 |
+
|
1488 |
+
# Other RVC stuff
|
1489 |
+
with gr.Row():
|
1490 |
+
sid0 = gr.Dropdown(label="1.Scegli il modello.", choices=sorted(names), value=check_for_name())
|
1491 |
+
refresh_button = gr.Button("Aggiorna", variant="primary")
|
1492 |
+
if check_for_name() != '':
|
1493 |
+
get_vc(sorted(names)[0])
|
1494 |
+
vc_transform0 = gr.Number(label="Pitch: 0 da uomo a uomo (o donna a donna); 12 da uomo a donna e -12 da donna a uomo", value=0)
|
1495 |
+
#clean_button = gr.Button(i18n("卸载音色省显存"), variant="primary")
|
1496 |
+
spk_item = gr.Slider(
|
1497 |
+
minimum=0,
|
1498 |
+
maximum=2333,
|
1499 |
+
step=1,
|
1500 |
+
label=i18n("请选择说话人id"),
|
1501 |
+
value=0,
|
1502 |
+
visible=False,
|
1503 |
+
interactive=True,
|
1504 |
+
)
|
1505 |
+
#clean_button.click(fn=clean, inputs=[], outputs=[sid0])
|
1506 |
+
sid0.change(
|
1507 |
+
fn=get_vc,
|
1508 |
+
inputs=[sid0],
|
1509 |
+
outputs=[spk_item],
|
1510 |
+
)
|
1511 |
+
but0 = gr.Button("Converti", variant="primary")
|
1512 |
+
with gr.Row():
|
1513 |
+
with gr.Column():
|
1514 |
+
with gr.Row():
|
1515 |
+
dropbox = gr.File(label="Inserisci il file audio e clicca il tasto aggiorna.")
|
1516 |
+
with gr.Row():
|
1517 |
+
record_button=gr.Audio(source="microphone", label="Oppure registra dal microfono.", type="filepath")
|
1518 |
+
with gr.Row():
|
1519 |
+
input_audio0 = gr.Dropdown(
|
1520 |
+
label="2.Scegli l'audio.",
|
1521 |
+
value="./audios/Test_Audio.mp3",
|
1522 |
+
choices=audio_files
|
1523 |
+
)
|
1524 |
+
dropbox.upload(fn=save_to_wav2, inputs=[dropbox], outputs=[input_audio0])
|
1525 |
+
dropbox.upload(fn=change_choices2, inputs=[], outputs=[input_audio0])
|
1526 |
+
refresh_button2 = gr.Button("Aggiorna", variant="primary", size='sm')
|
1527 |
+
record_button.change(fn=save_to_wav, inputs=[record_button], outputs=[input_audio0])
|
1528 |
+
record_button.change(fn=change_choices2, inputs=[], outputs=[input_audio0])
|
1529 |
+
with gr.Row():
|
1530 |
+
with gr.Accordion('Text To Speech', open=False):
|
1531 |
+
with gr.Column():
|
1532 |
+
lang = gr.Radio(label='Chinese & Japanese do not work with ElevenLabs currently.',choices=['en','es','fr','pt','zh-CN','de','hi','ja'], value='en')
|
1533 |
+
api_box = gr.Textbox(label="Enter your API Key for ElevenLabs, or leave empty to use GoogleTTS", value='')
|
1534 |
+
elevenid=gr.Dropdown(label="Voice:", choices=eleven_voices)
|
1535 |
+
with gr.Column():
|
1536 |
+
tfs = gr.Textbox(label="Input your Text", interactive=True, value="This is a test.")
|
1537 |
+
tts_button = gr.Button(value="Speak")
|
1538 |
+
tts_button.click(fn=elevenTTS, inputs=[api_box,tfs, elevenid, lang], outputs=[record_button, input_audio0])
|
1539 |
+
with gr.Row():
|
1540 |
+
with gr.Accordion('Wav2Lip', open=False):
|
1541 |
+
with gr.Row():
|
1542 |
+
size = gr.Radio(label='Resolution:',choices=['Half','Full'])
|
1543 |
+
face = gr.UploadButton("Upload A Character",type='file')
|
1544 |
+
faces = gr.Dropdown(label="OR Choose one:", choices=['None','Ben Shapiro','Andrew Tate'])
|
1545 |
+
with gr.Row():
|
1546 |
+
preview = gr.Textbox(label="Status:",interactive=False)
|
1547 |
+
face.upload(fn=success_message,inputs=[face], outputs=[preview, faces])
|
1548 |
+
with gr.Row():
|
1549 |
+
animation = gr.Video(type='filepath')
|
1550 |
+
refresh_button2.click(fn=change_choices2, inputs=[], outputs=[input_audio0, animation])
|
1551 |
+
with gr.Row():
|
1552 |
+
animate_button = gr.Button('Animate')
|
1553 |
+
|
1554 |
+
with gr.Column():
|
1555 |
+
with gr.Accordion("Impostazioni Index", open=False):
|
1556 |
+
file_index1 = gr.Dropdown(
|
1557 |
+
label="3. Scegli il file index (nel caso non sia stato fatto in automatico.)",
|
1558 |
+
choices=get_indexes(),
|
1559 |
+
value=get_index(),
|
1560 |
+
interactive=True,
|
1561 |
+
)
|
1562 |
+
sid0.change(fn=match_index, inputs=[sid0],outputs=[file_index1])
|
1563 |
+
refresh_button.click(
|
1564 |
+
fn=change_choices, inputs=[], outputs=[sid0, file_index1]
|
1565 |
+
)
|
1566 |
+
# file_big_npy1 = gr.Textbox(
|
1567 |
+
# label=i18n("特征文件路径"),
|
1568 |
+
# value="E:\\codes\py39\\vits_vc_gpu_train\\logs\\mi-test-1key\\total_fea.npy",
|
1569 |
+
# interactive=True,
|
1570 |
+
# )
|
1571 |
+
index_rate1 = gr.Slider(
|
1572 |
+
minimum=0,
|
1573 |
+
maximum=1,
|
1574 |
+
label=i18n("检索特征占比"),
|
1575 |
+
value=0.66,
|
1576 |
+
interactive=True,
|
1577 |
+
)
|
1578 |
+
vc_output2 = gr.Audio(
|
1579 |
+
label="Risultato finale (Click sui tre puntini per scaricare l'audio)",
|
1580 |
+
type='filepath',
|
1581 |
+
interactive=False,
|
1582 |
+
)
|
1583 |
+
animate_button.click(fn=mouth, inputs=[size, face, vc_output2, faces], outputs=[animation, preview])
|
1584 |
+
with gr.Accordion("Opzioni avanzate", open=False):
|
1585 |
+
f0method0 = gr.Radio(
|
1586 |
+
label="Opzionale: Cambia il meteodo di estrazione del pitch.\nIlaria consiglia di lasciare tutto così com'è.\nrmvpe è il miglior metodo, in caso di problemi potete provare anche mangio-crepe.",
|
1587 |
+
choices=["pm", "dio", "crepe-tiny", "mangio-crepe-tiny", "crepe", "harvest", "mangio-crepe", "rmvpe"], # Fork Feature. Add Crepe-Tiny
|
1588 |
+
value="rmvpe",
|
1589 |
+
interactive=True,
|
1590 |
+
)
|
1591 |
+
|
1592 |
+
crepe_hop_length = gr.Slider(
|
1593 |
+
minimum=1,
|
1594 |
+
maximum=512,
|
1595 |
+
step=1,
|
1596 |
+
label="Mangio-Crepe Hop Length. Higher numbers will reduce the chance of extreme pitch changes but lower numbers will increase accuracy. 64-192 is a good range to experiment with.",
|
1597 |
+
value=120,
|
1598 |
+
interactive=True,
|
1599 |
+
visible=False,
|
1600 |
+
)
|
1601 |
+
f0method0.change(fn=whethercrepeornah, inputs=[f0method0], outputs=[crepe_hop_length])
|
1602 |
+
filter_radius0 = gr.Slider(
|
1603 |
+
minimum=0,
|
1604 |
+
maximum=7,
|
1605 |
+
label=i18n(">=3则使用对harvest音高识别的结果使用中值滤波,数值为滤波半径,使用可以削弱哑音"),
|
1606 |
+
value=3,
|
1607 |
+
step=1,
|
1608 |
+
interactive=True,
|
1609 |
+
)
|
1610 |
+
resample_sr0 = gr.Slider(
|
1611 |
+
minimum=0,
|
1612 |
+
maximum=48000,
|
1613 |
+
label=i18n("后处理重采样至最终采样率,0为不进行重采样"),
|
1614 |
+
value=0,
|
1615 |
+
step=1,
|
1616 |
+
interactive=True,
|
1617 |
+
visible=False
|
1618 |
+
)
|
1619 |
+
rms_mix_rate0 = gr.Slider(
|
1620 |
+
minimum=0,
|
1621 |
+
maximum=1,
|
1622 |
+
label=i18n("输入源音量包络替换输出音量包络融合比例,越靠近1越使用输出包络"),
|
1623 |
+
value=0.21,
|
1624 |
+
interactive=True,
|
1625 |
+
)
|
1626 |
+
protect0 = gr.Slider(
|
1627 |
+
minimum=0,
|
1628 |
+
maximum=0.5,
|
1629 |
+
label=i18n("保护清辅音和呼吸声,防止电音撕裂等artifact,拉满0.5不开启,调低加大保护力度但可能降低索引效果"),
|
1630 |
+
value=0.33,
|
1631 |
+
step=0.01,
|
1632 |
+
interactive=True,
|
1633 |
+
)
|
1634 |
+
formanting = gr.Checkbox(
|
1635 |
+
value=bool(DoFormant),
|
1636 |
+
label="[EXPERIMENTAL] Formant shift inference audio",
|
1637 |
+
info="Used for male to female and vice-versa conversions",
|
1638 |
+
interactive=True,
|
1639 |
+
visible=True,
|
1640 |
+
)
|
1641 |
+
|
1642 |
+
formant_preset = gr.Dropdown(
|
1643 |
+
value='',
|
1644 |
+
choices=get_fshift_presets(),
|
1645 |
+
label="browse presets for formanting",
|
1646 |
+
visible=bool(DoFormant),
|
1647 |
+
)
|
1648 |
+
formant_refresh_button = gr.Button(
|
1649 |
+
value='\U0001f504',
|
1650 |
+
visible=bool(DoFormant),
|
1651 |
+
variant='primary',
|
1652 |
+
)
|
1653 |
+
#formant_refresh_button = ToolButton( elem_id='1')
|
1654 |
+
#create_refresh_button(formant_preset, lambda: {"choices": formant_preset}, "refresh_list_shiftpresets")
|
1655 |
+
|
1656 |
+
qfrency = gr.Slider(
|
1657 |
+
value=Quefrency,
|
1658 |
+
info="Default value is 1.0",
|
1659 |
+
label="Quefrency for formant shifting",
|
1660 |
+
minimum=0.0,
|
1661 |
+
maximum=16.0,
|
1662 |
+
step=0.1,
|
1663 |
+
visible=bool(DoFormant),
|
1664 |
+
interactive=True,
|
1665 |
+
)
|
1666 |
+
tmbre = gr.Slider(
|
1667 |
+
value=Timbre,
|
1668 |
+
info="Default value is 1.0",
|
1669 |
+
label="Timbre for formant shifting",
|
1670 |
+
minimum=0.0,
|
1671 |
+
maximum=16.0,
|
1672 |
+
step=0.1,
|
1673 |
+
visible=bool(DoFormant),
|
1674 |
+
interactive=True,
|
1675 |
+
)
|
1676 |
+
|
1677 |
+
formant_preset.change(fn=preset_apply, inputs=[formant_preset, qfrency, tmbre], outputs=[qfrency, tmbre])
|
1678 |
+
frmntbut = gr.Button("Apply", variant="primary", visible=bool(DoFormant))
|
1679 |
+
formanting.change(fn=formant_enabled,inputs=[formanting,qfrency,tmbre,frmntbut,formant_preset,formant_refresh_button],outputs=[formanting,qfrency,tmbre,frmntbut,formant_preset,formant_refresh_button])
|
1680 |
+
frmntbut.click(fn=formant_apply,inputs=[qfrency, tmbre], outputs=[qfrency, tmbre])
|
1681 |
+
formant_refresh_button.click(fn=update_fshift_presets,inputs=[formant_preset, qfrency, tmbre],outputs=[formant_preset, qfrency, tmbre])
|
1682 |
+
with gr.Row():
|
1683 |
+
vc_output1 = gr.Textbox("")
|
1684 |
+
f0_file = gr.File(label=i18n("F0曲线文件, 可选, 一行一个音高, 代替默认F0及升降调"), visible=False)
|
1685 |
+
|
1686 |
+
but0.click(
|
1687 |
+
vc_single,
|
1688 |
+
[
|
1689 |
+
spk_item,
|
1690 |
+
input_audio0,
|
1691 |
+
vc_transform0,
|
1692 |
+
f0_file,
|
1693 |
+
f0method0,
|
1694 |
+
file_index1,
|
1695 |
+
# file_index2,
|
1696 |
+
# file_big_npy1,
|
1697 |
+
index_rate1,
|
1698 |
+
filter_radius0,
|
1699 |
+
resample_sr0,
|
1700 |
+
rms_mix_rate0,
|
1701 |
+
protect0,
|
1702 |
+
crepe_hop_length
|
1703 |
+
],
|
1704 |
+
[vc_output1, vc_output2],
|
1705 |
+
)
|
1706 |
+
|
1707 |
+
with gr.Accordion("Batch Conversion",open=False):
|
1708 |
+
with gr.Row():
|
1709 |
+
with gr.Column():
|
1710 |
+
vc_transform1 = gr.Number(
|
1711 |
+
label=i18n("变调(整数, 半音数量, 升八度12降八度-12)"), value=0
|
1712 |
+
)
|
1713 |
+
opt_input = gr.Textbox(label=i18n("指定输出文件夹"), value="opt")
|
1714 |
+
f0method1 = gr.Radio(
|
1715 |
+
label=i18n(
|
1716 |
+
"选择音高提取算法,输入歌声可用pm提速,harvest低音好但巨慢无比,crepe效果好但吃GPU"
|
1717 |
+
),
|
1718 |
+
choices=["pm", "harvest", "crepe", "rmvpe"],
|
1719 |
+
value="rmvpe",
|
1720 |
+
interactive=True,
|
1721 |
+
)
|
1722 |
+
filter_radius1 = gr.Slider(
|
1723 |
+
minimum=0,
|
1724 |
+
maximum=7,
|
1725 |
+
label=i18n(">=3则使用对harvest音高识别的结果使用中值滤波,数值为滤波半径,使用可以削弱哑音"),
|
1726 |
+
value=3,
|
1727 |
+
step=1,
|
1728 |
+
interactive=True,
|
1729 |
+
)
|
1730 |
+
with gr.Column():
|
1731 |
+
file_index3 = gr.Textbox(
|
1732 |
+
label=i18n("特征检索库文件路径,为空则使用下拉的选择结果"),
|
1733 |
+
value="",
|
1734 |
+
interactive=True,
|
1735 |
+
)
|
1736 |
+
file_index4 = gr.Dropdown(
|
1737 |
+
label=i18n("自动检测index路径,下拉式选择(dropdown)"),
|
1738 |
+
choices=sorted(index_paths),
|
1739 |
+
interactive=True,
|
1740 |
+
)
|
1741 |
+
refresh_button.click(
|
1742 |
+
fn=lambda: change_choices()[1],
|
1743 |
+
inputs=[],
|
1744 |
+
outputs=file_index4,
|
1745 |
+
)
|
1746 |
+
# file_big_npy2 = gr.Textbox(
|
1747 |
+
# label=i18n("特征文件路径"),
|
1748 |
+
# value="E:\\codes\\py39\\vits_vc_gpu_train\\logs\\mi-test-1key\\total_fea.npy",
|
1749 |
+
# interactive=True,
|
1750 |
+
# )
|
1751 |
+
index_rate2 = gr.Slider(
|
1752 |
+
minimum=0,
|
1753 |
+
maximum=1,
|
1754 |
+
label=i18n("检索特征占比"),
|
1755 |
+
value=1,
|
1756 |
+
interactive=True,
|
1757 |
+
)
|
1758 |
+
with gr.Column():
|
1759 |
+
resample_sr1 = gr.Slider(
|
1760 |
+
minimum=0,
|
1761 |
+
maximum=48000,
|
1762 |
+
label=i18n("后处理重采样至最终采样率,0为不进行重采样"),
|
1763 |
+
value=0,
|
1764 |
+
step=1,
|
1765 |
+
interactive=True,
|
1766 |
+
)
|
1767 |
+
rms_mix_rate1 = gr.Slider(
|
1768 |
+
minimum=0,
|
1769 |
+
maximum=1,
|
1770 |
+
label=i18n("输入源音量包络替换输出音量包络融合比例,越靠近1越使用输出包络"),
|
1771 |
+
value=1,
|
1772 |
+
interactive=True,
|
1773 |
+
)
|
1774 |
+
protect1 = gr.Slider(
|
1775 |
+
minimum=0,
|
1776 |
+
maximum=0.5,
|
1777 |
+
label=i18n(
|
1778 |
+
"保护清辅音和呼吸声,防止电音撕裂等artifact,拉满0.5不开启,调低加大保护力度但可能降低索引效果"
|
1779 |
+
),
|
1780 |
+
value=0.33,
|
1781 |
+
step=0.01,
|
1782 |
+
interactive=True,
|
1783 |
+
)
|
1784 |
+
with gr.Column():
|
1785 |
+
dir_input = gr.Textbox(
|
1786 |
+
label=i18n("输入待处理音频文件夹路径(去文件管理器地址栏拷就行了)"),
|
1787 |
+
value="E:\codes\py39\\test-20230416b\\todo-songs",
|
1788 |
+
)
|
1789 |
+
inputs = gr.File(
|
1790 |
+
file_count="multiple", label=i18n("也可��量输入音频文件, 二选一, 优先读文件夹")
|
1791 |
+
)
|
1792 |
+
with gr.Row():
|
1793 |
+
format1 = gr.Radio(
|
1794 |
+
label=i18n("导出文件格式"),
|
1795 |
+
choices=["wav", "flac", "mp3", "m4a"],
|
1796 |
+
value="flac",
|
1797 |
+
interactive=True,
|
1798 |
+
)
|
1799 |
+
but1 = gr.Button(i18n("转换"), variant="primary")
|
1800 |
+
vc_output3 = gr.Textbox(label=i18n("输出信息"))
|
1801 |
+
but1.click(
|
1802 |
+
vc_multi,
|
1803 |
+
[
|
1804 |
+
spk_item,
|
1805 |
+
dir_input,
|
1806 |
+
opt_input,
|
1807 |
+
inputs,
|
1808 |
+
vc_transform1,
|
1809 |
+
f0method1,
|
1810 |
+
file_index3,
|
1811 |
+
file_index4,
|
1812 |
+
# file_big_npy2,
|
1813 |
+
index_rate2,
|
1814 |
+
filter_radius1,
|
1815 |
+
resample_sr1,
|
1816 |
+
rms_mix_rate1,
|
1817 |
+
protect1,
|
1818 |
+
format1,
|
1819 |
+
crepe_hop_length,
|
1820 |
+
],
|
1821 |
+
[vc_output3],
|
1822 |
+
)
|
1823 |
+
but1.click(fn=lambda: easy_uploader.clear())
|
1824 |
+
with gr.TabItem("Scarica modelli"):
|
1825 |
+
with gr.Row():
|
1826 |
+
url=gr.Textbox(label="Inserisci l'URL del modello:")
|
1827 |
+
with gr.Row():
|
1828 |
+
model = gr.Textbox(label="Nome del modello (senza spazi):")
|
1829 |
+
download_button=gr.Button("Scarica")
|
1830 |
+
with gr.Row():
|
1831 |
+
status_bar=gr.Textbox(label="")
|
1832 |
+
download_button.click(fn=download_from_url, inputs=[url, model], outputs=[status_bar])
|
1833 |
+
with gr.Row():
|
1834 |
+
gr.Markdown(
|
1835 |
+
"""
|
1836 |
+
Made with 💖 by Ilaria | Sostieni Ilaria su [Ko-Fi](https://ko-fi.com/ilariaowo)
|
1837 |
+
"""
|
1838 |
+
)
|
1839 |
+
|
1840 |
+
def has_two_files_in_pretrained_folder():
|
1841 |
+
pretrained_folder = "./pretrained/"
|
1842 |
+
if not os.path.exists(pretrained_folder):
|
1843 |
+
return False
|
1844 |
+
|
1845 |
+
files_in_folder = os.listdir(pretrained_folder)
|
1846 |
+
num_files = len(files_in_folder)
|
1847 |
+
return num_files >= 2
|
1848 |
+
|
1849 |
+
if has_two_files_in_pretrained_folder():
|
1850 |
+
print("Pretrained weights are downloaded. Training tab enabled!\n-------------------------------")
|
1851 |
+
with gr.TabItem("Train", visible=False):
|
1852 |
+
with gr.Row():
|
1853 |
+
with gr.Column():
|
1854 |
+
exp_dir1 = gr.Textbox(label="Voice Name:", value="My-Voice")
|
1855 |
+
sr2 = gr.Radio(
|
1856 |
+
label=i18n("目标采样率"),
|
1857 |
+
choices=["40k", "48k"],
|
1858 |
+
value="40k",
|
1859 |
+
interactive=True,
|
1860 |
+
visible=False
|
1861 |
+
)
|
1862 |
+
if_f0_3 = gr.Radio(
|
1863 |
+
label=i18n("模型是否带音高指导(唱歌一定要, 语音可以不要)"),
|
1864 |
+
choices=[True, False],
|
1865 |
+
value=True,
|
1866 |
+
interactive=True,
|
1867 |
+
visible=False
|
1868 |
+
)
|
1869 |
+
version19 = gr.Radio(
|
1870 |
+
label="RVC version",
|
1871 |
+
choices=["v1", "v2"],
|
1872 |
+
value="v2",
|
1873 |
+
interactive=True,
|
1874 |
+
visible=False,
|
1875 |
+
)
|
1876 |
+
np7 = gr.Slider(
|
1877 |
+
minimum=0,
|
1878 |
+
maximum=config.n_cpu,
|
1879 |
+
step=1,
|
1880 |
+
label="# of CPUs for data processing (Leave as it is)",
|
1881 |
+
value=config.n_cpu,
|
1882 |
+
interactive=True,
|
1883 |
+
visible=True
|
1884 |
+
)
|
1885 |
+
trainset_dir4 = gr.Textbox(label="Path to your dataset (audios, not zip):", value="./dataset")
|
1886 |
+
easy_uploader = gr.Files(label='OR Drop your audios here. They will be uploaded in your dataset path above.',file_types=['audio'])
|
1887 |
+
but1 = gr.Button("1. Process The Dataset", variant="primary")
|
1888 |
+
info1 = gr.Textbox(label="Status (wait until it says 'end preprocess'):", value="")
|
1889 |
+
easy_uploader.upload(fn=upload_to_dataset, inputs=[easy_uploader, trainset_dir4], outputs=[info1])
|
1890 |
+
but1.click(
|
1891 |
+
preprocess_dataset, [trainset_dir4, exp_dir1, sr2, np7], [info1]
|
1892 |
+
)
|
1893 |
+
with gr.Column():
|
1894 |
+
spk_id5 = gr.Slider(
|
1895 |
+
minimum=0,
|
1896 |
+
maximum=4,
|
1897 |
+
step=1,
|
1898 |
+
label=i18n("请指定说话人id"),
|
1899 |
+
value=0,
|
1900 |
+
interactive=True,
|
1901 |
+
visible=False
|
1902 |
+
)
|
1903 |
+
with gr.Accordion('GPU Settings', open=False, visible=False):
|
1904 |
+
gpus6 = gr.Textbox(
|
1905 |
+
label=i18n("以-分隔输入使用的卡号, 例如 0-1-2 使用卡0和卡1和卡2"),
|
1906 |
+
value=gpus,
|
1907 |
+
interactive=True,
|
1908 |
+
visible=False
|
1909 |
+
)
|
1910 |
+
gpu_info9 = gr.Textbox(label=i18n("显卡信息"), value=gpu_info)
|
1911 |
+
f0method8 = gr.Radio(
|
1912 |
+
label=i18n(
|
1913 |
+
"选择音高提取算法:输入歌声可用pm提速,高质量语音但CPU差可用dio提速,harvest质量更好但慢"
|
1914 |
+
),
|
1915 |
+
choices=["harvest","crepe", "mangio-crepe", "rmvpe"], # Fork feature: Crepe on f0 extraction for training.
|
1916 |
+
value="rmvpe",
|
1917 |
+
interactive=True,
|
1918 |
+
)
|
1919 |
+
|
1920 |
+
extraction_crepe_hop_length = gr.Slider(
|
1921 |
+
minimum=1,
|
1922 |
+
maximum=512,
|
1923 |
+
step=1,
|
1924 |
+
label=i18n("crepe_hop_length"),
|
1925 |
+
value=128,
|
1926 |
+
interactive=True,
|
1927 |
+
visible=False,
|
1928 |
+
)
|
1929 |
+
f0method8.change(fn=whethercrepeornah, inputs=[f0method8], outputs=[extraction_crepe_hop_length])
|
1930 |
+
but2 = gr.Button("2. Pitch Extraction", variant="primary")
|
1931 |
+
info2 = gr.Textbox(label="Status(Check the Colab Notebook's cell output):", value="", max_lines=8)
|
1932 |
+
but2.click(
|
1933 |
+
extract_f0_feature,
|
1934 |
+
[gpus6, np7, f0method8, if_f0_3, exp_dir1, version19, extraction_crepe_hop_length],
|
1935 |
+
[info2],
|
1936 |
+
)
|
1937 |
+
with gr.Row():
|
1938 |
+
with gr.Column():
|
1939 |
+
total_epoch11 = gr.Slider(
|
1940 |
+
minimum=1,
|
1941 |
+
maximum=5000,
|
1942 |
+
step=10,
|
1943 |
+
label="Total # of training epochs (IF you choose a value too high, your model will sound horribly overtrained.):",
|
1944 |
+
value=250,
|
1945 |
+
interactive=True,
|
1946 |
+
)
|
1947 |
+
butstop = gr.Button(
|
1948 |
+
"Stop Training",
|
1949 |
+
variant='primary',
|
1950 |
+
visible=False,
|
1951 |
+
)
|
1952 |
+
but3 = gr.Button("3. Train Model", variant="primary", visible=True)
|
1953 |
+
|
1954 |
+
but3.click(fn=stoptraining, inputs=[gr.Number(value=0, visible=False)], outputs=[but3, butstop])
|
1955 |
+
butstop.click(fn=stoptraining, inputs=[gr.Number(value=1, visible=False)], outputs=[butstop, but3])
|
1956 |
+
|
1957 |
+
|
1958 |
+
but4 = gr.Button("4.Train Index", variant="primary")
|
1959 |
+
info3 = gr.Textbox(label="Status(Check the Colab Notebook's cell output):", value="", max_lines=10)
|
1960 |
+
with gr.Accordion("Training Preferences (You can leave these as they are)", open=False):
|
1961 |
+
#gr.Markdown(value=i18n("step3: 填写训练设置, 开始训练模型和索引"))
|
1962 |
+
with gr.Column():
|
1963 |
+
save_epoch10 = gr.Slider(
|
1964 |
+
minimum=1,
|
1965 |
+
maximum=200,
|
1966 |
+
step=1,
|
1967 |
+
label="Backup every X amount of epochs:",
|
1968 |
+
value=10,
|
1969 |
+
interactive=True,
|
1970 |
+
)
|
1971 |
+
batch_size12 = gr.Slider(
|
1972 |
+
minimum=1,
|
1973 |
+
maximum=40,
|
1974 |
+
step=1,
|
1975 |
+
label="Batch Size (LEAVE IT unless you know what you're doing!):",
|
1976 |
+
value=default_batch_size,
|
1977 |
+
interactive=True,
|
1978 |
+
)
|
1979 |
+
if_save_latest13 = gr.Checkbox(
|
1980 |
+
label="Save only the latest '.ckpt' file to save disk space.",
|
1981 |
+
value=True,
|
1982 |
+
interactive=True,
|
1983 |
+
)
|
1984 |
+
if_cache_gpu17 = gr.Checkbox(
|
1985 |
+
label="Cache all training sets to GPU memory. Caching small datasets (less than 10 minutes) can speed up training, but caching large datasets will consume a lot of GPU memory and may not provide much speed improvement.",
|
1986 |
+
value=False,
|
1987 |
+
interactive=True,
|
1988 |
+
)
|
1989 |
+
if_save_every_weights18 = gr.Checkbox(
|
1990 |
+
label="Save a small final model to the 'weights' folder at each save point.",
|
1991 |
+
value=True,
|
1992 |
+
interactive=True,
|
1993 |
+
)
|
1994 |
+
zip_model = gr.Button('5. Download Model')
|
1995 |
+
zipped_model = gr.Files(label='Your Model and Index file can be downloaded here:')
|
1996 |
+
zip_model.click(fn=zip_downloader, inputs=[exp_dir1], outputs=[zipped_model, info3])
|
1997 |
+
with gr.Group():
|
1998 |
+
with gr.Accordion("Base Model Locations:", open=False, visible=False):
|
1999 |
+
pretrained_G14 = gr.Textbox(
|
2000 |
+
label=i18n("加载预训练底模G路径"),
|
2001 |
+
value="pretrained_v2/f0G40k.pth",
|
2002 |
+
interactive=True,
|
2003 |
+
)
|
2004 |
+
pretrained_D15 = gr.Textbox(
|
2005 |
+
label=i18n("加载预训练底模D路径"),
|
2006 |
+
value="pretrained_v2/f0D40k.pth",
|
2007 |
+
interactive=True,
|
2008 |
+
)
|
2009 |
+
gpus16 = gr.Textbox(
|
2010 |
+
label=i18n("以-分隔输入使用的卡号, 例如 0-1-2 使用卡0和卡1和卡2"),
|
2011 |
+
value=gpus,
|
2012 |
+
interactive=True,
|
2013 |
+
)
|
2014 |
+
sr2.change(
|
2015 |
+
change_sr2,
|
2016 |
+
[sr2, if_f0_3, version19],
|
2017 |
+
[pretrained_G14, pretrained_D15, version19],
|
2018 |
+
)
|
2019 |
+
version19.change(
|
2020 |
+
change_version19,
|
2021 |
+
[sr2, if_f0_3, version19],
|
2022 |
+
[pretrained_G14, pretrained_D15],
|
2023 |
+
)
|
2024 |
+
if_f0_3.change(
|
2025 |
+
change_f0,
|
2026 |
+
[if_f0_3, sr2, version19],
|
2027 |
+
[f0method8, pretrained_G14, pretrained_D15],
|
2028 |
+
)
|
2029 |
+
but5 = gr.Button(i18n("一键训练"), variant="primary", visible=False)
|
2030 |
+
but3.click(
|
2031 |
+
click_train,
|
2032 |
+
[
|
2033 |
+
exp_dir1,
|
2034 |
+
sr2,
|
2035 |
+
if_f0_3,
|
2036 |
+
spk_id5,
|
2037 |
+
save_epoch10,
|
2038 |
+
total_epoch11,
|
2039 |
+
batch_size12,
|
2040 |
+
if_save_latest13,
|
2041 |
+
pretrained_G14,
|
2042 |
+
pretrained_D15,
|
2043 |
+
gpus16,
|
2044 |
+
if_cache_gpu17,
|
2045 |
+
if_save_every_weights18,
|
2046 |
+
version19,
|
2047 |
+
],
|
2048 |
+
[
|
2049 |
+
info3,
|
2050 |
+
butstop,
|
2051 |
+
but3,
|
2052 |
+
],
|
2053 |
+
)
|
2054 |
+
but4.click(train_index, [exp_dir1, version19], info3)
|
2055 |
+
but5.click(
|
2056 |
+
train1key,
|
2057 |
+
[
|
2058 |
+
exp_dir1,
|
2059 |
+
sr2,
|
2060 |
+
if_f0_3,
|
2061 |
+
trainset_dir4,
|
2062 |
+
spk_id5,
|
2063 |
+
np7,
|
2064 |
+
f0method8,
|
2065 |
+
save_epoch10,
|
2066 |
+
total_epoch11,
|
2067 |
+
batch_size12,
|
2068 |
+
if_save_latest13,
|
2069 |
+
pretrained_G14,
|
2070 |
+
pretrained_D15,
|
2071 |
+
gpus16,
|
2072 |
+
if_cache_gpu17,
|
2073 |
+
if_save_every_weights18,
|
2074 |
+
version19,
|
2075 |
+
extraction_crepe_hop_length
|
2076 |
+
],
|
2077 |
+
info3,
|
2078 |
+
)
|
2079 |
+
|
2080 |
+
else:
|
2081 |
+
print(
|
2082 |
+
"Pretrained weights not downloaded. Disabling training tab.\n"
|
2083 |
+
"Wondering how to train a voice? Visit here for the RVC model training guide: https://t.ly/RVC_Training_Guide\n"
|
2084 |
+
"-------------------------------\n"
|
2085 |
+
)
|
2086 |
+
|
2087 |
+
app.queue(concurrency_count=511, max_size=1022).launch(share=False, quiet=True)
|
2088 |
+
#endregion
|
audios/Test_Audio.mp3
ADDED
Binary file (79.4 kB). View file
|
|
audios/gitignore.txt
ADDED
File without changes
|
config.py
ADDED
@@ -0,0 +1,204 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import argparse
|
2 |
+
import sys
|
3 |
+
import torch
|
4 |
+
import json
|
5 |
+
from multiprocessing import cpu_count
|
6 |
+
|
7 |
+
global usefp16
|
8 |
+
usefp16 = False
|
9 |
+
|
10 |
+
|
11 |
+
def use_fp32_config():
|
12 |
+
usefp16 = False
|
13 |
+
device_capability = 0
|
14 |
+
if torch.cuda.is_available():
|
15 |
+
device = torch.device("cuda:0") # Assuming you have only one GPU (index 0).
|
16 |
+
device_capability = torch.cuda.get_device_capability(device)[0]
|
17 |
+
if device_capability >= 7:
|
18 |
+
usefp16 = True
|
19 |
+
for config_file in ["32k.json", "40k.json", "48k.json"]:
|
20 |
+
with open(f"configs/{config_file}", "r") as d:
|
21 |
+
data = json.load(d)
|
22 |
+
|
23 |
+
if "train" in data and "fp16_run" in data["train"]:
|
24 |
+
data["train"]["fp16_run"] = True
|
25 |
+
|
26 |
+
with open(f"configs/{config_file}", "w") as d:
|
27 |
+
json.dump(data, d, indent=4)
|
28 |
+
|
29 |
+
print(f"Set fp16_run to true in {config_file}")
|
30 |
+
|
31 |
+
with open(
|
32 |
+
"trainset_preprocess_pipeline_print.py", "r", encoding="utf-8"
|
33 |
+
) as f:
|
34 |
+
strr = f.read()
|
35 |
+
|
36 |
+
strr = strr.replace("3.0", "3.7")
|
37 |
+
|
38 |
+
with open(
|
39 |
+
"trainset_preprocess_pipeline_print.py", "w", encoding="utf-8"
|
40 |
+
) as f:
|
41 |
+
f.write(strr)
|
42 |
+
else:
|
43 |
+
for config_file in ["32k.json", "40k.json", "48k.json"]:
|
44 |
+
with open(f"configs/{config_file}", "r") as f:
|
45 |
+
data = json.load(f)
|
46 |
+
|
47 |
+
if "train" in data and "fp16_run" in data["train"]:
|
48 |
+
data["train"]["fp16_run"] = False
|
49 |
+
|
50 |
+
with open(f"configs/{config_file}", "w") as d:
|
51 |
+
json.dump(data, d, indent=4)
|
52 |
+
|
53 |
+
print(f"Set fp16_run to false in {config_file}")
|
54 |
+
|
55 |
+
with open(
|
56 |
+
"trainset_preprocess_pipeline_print.py", "r", encoding="utf-8"
|
57 |
+
) as f:
|
58 |
+
strr = f.read()
|
59 |
+
|
60 |
+
strr = strr.replace("3.7", "3.0")
|
61 |
+
|
62 |
+
with open(
|
63 |
+
"trainset_preprocess_pipeline_print.py", "w", encoding="utf-8"
|
64 |
+
) as f:
|
65 |
+
f.write(strr)
|
66 |
+
else:
|
67 |
+
print(
|
68 |
+
"CUDA is not available. Make sure you have an NVIDIA GPU and CUDA installed."
|
69 |
+
)
|
70 |
+
return (usefp16, device_capability)
|
71 |
+
|
72 |
+
|
73 |
+
class Config:
|
74 |
+
def __init__(self):
|
75 |
+
self.device = "cuda:0"
|
76 |
+
self.is_half = True
|
77 |
+
self.n_cpu = 0
|
78 |
+
self.gpu_name = None
|
79 |
+
self.gpu_mem = None
|
80 |
+
(
|
81 |
+
self.python_cmd,
|
82 |
+
self.listen_port,
|
83 |
+
self.iscolab,
|
84 |
+
self.noparallel,
|
85 |
+
self.noautoopen,
|
86 |
+
self.paperspace,
|
87 |
+
self.is_cli,
|
88 |
+
) = self.arg_parse()
|
89 |
+
|
90 |
+
self.x_pad, self.x_query, self.x_center, self.x_max = self.device_config()
|
91 |
+
|
92 |
+
@staticmethod
|
93 |
+
def arg_parse() -> tuple:
|
94 |
+
exe = sys.executable or "python"
|
95 |
+
parser = argparse.ArgumentParser()
|
96 |
+
parser.add_argument("--port", type=int, default=7865, help="Listen port")
|
97 |
+
parser.add_argument("--pycmd", type=str, default=exe, help="Python command")
|
98 |
+
parser.add_argument("--colab", action="store_true", help="Launch in colab")
|
99 |
+
parser.add_argument(
|
100 |
+
"--noparallel", action="store_true", help="Disable parallel processing"
|
101 |
+
)
|
102 |
+
parser.add_argument(
|
103 |
+
"--noautoopen",
|
104 |
+
action="store_true",
|
105 |
+
help="Do not open in browser automatically",
|
106 |
+
)
|
107 |
+
parser.add_argument( # Fork Feature. Paperspace integration for web UI
|
108 |
+
"--paperspace",
|
109 |
+
action="store_true",
|
110 |
+
help="Note that this argument just shares a gradio link for the web UI. Thus can be used on other non-local CLI systems.",
|
111 |
+
)
|
112 |
+
parser.add_argument( # Fork Feature. Embed a CLI into the infer-web.py
|
113 |
+
"--is_cli",
|
114 |
+
action="store_true",
|
115 |
+
help="Use the CLI instead of setting up a gradio UI. This flag will launch an RVC text interface where you can execute functions from infer-web.py!",
|
116 |
+
)
|
117 |
+
cmd_opts = parser.parse_args()
|
118 |
+
|
119 |
+
cmd_opts.port = cmd_opts.port if 0 <= cmd_opts.port <= 65535 else 7865
|
120 |
+
|
121 |
+
return (
|
122 |
+
cmd_opts.pycmd,
|
123 |
+
cmd_opts.port,
|
124 |
+
cmd_opts.colab,
|
125 |
+
cmd_opts.noparallel,
|
126 |
+
cmd_opts.noautoopen,
|
127 |
+
cmd_opts.paperspace,
|
128 |
+
cmd_opts.is_cli,
|
129 |
+
)
|
130 |
+
|
131 |
+
# has_mps is only available in nightly pytorch (for now) and MasOS 12.3+.
|
132 |
+
# check `getattr` and try it for compatibility
|
133 |
+
@staticmethod
|
134 |
+
def has_mps() -> bool:
|
135 |
+
if not torch.backends.mps.is_available():
|
136 |
+
return False
|
137 |
+
try:
|
138 |
+
torch.zeros(1).to(torch.device("mps"))
|
139 |
+
return True
|
140 |
+
except Exception:
|
141 |
+
return False
|
142 |
+
|
143 |
+
def device_config(self) -> tuple:
|
144 |
+
if torch.cuda.is_available():
|
145 |
+
i_device = int(self.device.split(":")[-1])
|
146 |
+
self.gpu_name = torch.cuda.get_device_name(i_device)
|
147 |
+
if (
|
148 |
+
("16" in self.gpu_name and "V100" not in self.gpu_name.upper())
|
149 |
+
or "P40" in self.gpu_name.upper()
|
150 |
+
or "1060" in self.gpu_name
|
151 |
+
or "1070" in self.gpu_name
|
152 |
+
or "1080" in self.gpu_name
|
153 |
+
):
|
154 |
+
print("Found GPU", self.gpu_name, ", force to fp32")
|
155 |
+
self.is_half = False
|
156 |
+
else:
|
157 |
+
print("Found GPU", self.gpu_name)
|
158 |
+
use_fp32_config()
|
159 |
+
self.gpu_mem = int(
|
160 |
+
torch.cuda.get_device_properties(i_device).total_memory
|
161 |
+
/ 1024
|
162 |
+
/ 1024
|
163 |
+
/ 1024
|
164 |
+
+ 0.4
|
165 |
+
)
|
166 |
+
if self.gpu_mem <= 4:
|
167 |
+
with open("trainset_preprocess_pipeline_print.py", "r") as f:
|
168 |
+
strr = f.read().replace("3.7", "3.0")
|
169 |
+
with open("trainset_preprocess_pipeline_print.py", "w") as f:
|
170 |
+
f.write(strr)
|
171 |
+
elif self.has_mps():
|
172 |
+
print("No supported Nvidia GPU found, use MPS instead")
|
173 |
+
self.device = "mps"
|
174 |
+
self.is_half = False
|
175 |
+
use_fp32_config()
|
176 |
+
else:
|
177 |
+
print("No supported Nvidia GPU found, use CPU instead")
|
178 |
+
self.device = "cpu"
|
179 |
+
self.is_half = False
|
180 |
+
use_fp32_config()
|
181 |
+
|
182 |
+
if self.n_cpu == 0:
|
183 |
+
self.n_cpu = cpu_count()
|
184 |
+
|
185 |
+
if self.is_half:
|
186 |
+
# 6G显存配置
|
187 |
+
x_pad = 3
|
188 |
+
x_query = 10
|
189 |
+
x_center = 60
|
190 |
+
x_max = 65
|
191 |
+
else:
|
192 |
+
# 5G显存配置
|
193 |
+
x_pad = 1
|
194 |
+
x_query = 6
|
195 |
+
x_center = 38
|
196 |
+
x_max = 41
|
197 |
+
|
198 |
+
if self.gpu_mem != None and self.gpu_mem <= 4:
|
199 |
+
x_pad = 1
|
200 |
+
x_query = 5
|
201 |
+
x_center = 30
|
202 |
+
x_max = 32
|
203 |
+
|
204 |
+
return x_pad, x_query, x_center, x_max
|
gitattributes.txt
ADDED
@@ -0,0 +1,35 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
5 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
6 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
7 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
8 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
9 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
10 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
11 |
+
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
12 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
13 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
14 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
15 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
16 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
17 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
18 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
20 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
21 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
22 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
23 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
24 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
25 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
26 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
27 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
28 |
+
*.tar filter=lfs diff=lfs merge=lfs -text
|
29 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
30 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
31 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
32 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
33 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
gitignore.txt
ADDED
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
__pycache__/
|
2 |
+
weights/
|
3 |
+
TEMP/
|
4 |
+
logs/
|
5 |
+
csvdb/
|
6 |
+
|
7 |
+
# Environment
|
8 |
+
venv/
|
9 |
+
|
10 |
+
# Models
|
11 |
+
hubert_base.pt
|
12 |
+
rmvpe.pt
|
i18n.py
ADDED
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import locale
|
2 |
+
import json
|
3 |
+
import os
|
4 |
+
|
5 |
+
|
6 |
+
def load_language_list(language):
|
7 |
+
with open(f"./i18n/{language}.json", "r", encoding="utf-8") as f:
|
8 |
+
language_list = json.load(f)
|
9 |
+
return language_list
|
10 |
+
|
11 |
+
|
12 |
+
class I18nAuto:
|
13 |
+
def __init__(self, language=None):
|
14 |
+
if language in ["Auto", None]:
|
15 |
+
language = locale.getdefaultlocale()[
|
16 |
+
0
|
17 |
+
] # getlocale can't identify the system's language ((None, None))
|
18 |
+
if not os.path.exists(f"./i18n/{language}.json"):
|
19 |
+
language = "it_IT"
|
20 |
+
self.language = language
|
21 |
+
# print("Use Language:", language)
|
22 |
+
self.language_map = load_language_list(language)
|
23 |
+
|
24 |
+
def __call__(self, key):
|
25 |
+
return self.language_map.get(key, key)
|
26 |
+
|
27 |
+
def print(self):
|
28 |
+
print("Use Language:", self.language)
|
i18n/en_US.json
ADDED
@@ -0,0 +1,130 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"很遗憾您这没有能用的显卡来支持您训练": "Sfortunatamente, non è disponibile alcuna GPU compatibile per supportare l'addestramento.",
|
3 |
+
"是": "SÌ",
|
4 |
+
"step1:正在处理数据": "Passaggio 1: elaborazione dei dati",
|
5 |
+
"step2a:无需提取音高": "Step 2a: Saltare l'estrazione del tono",
|
6 |
+
"step2b:正在提取特征": "Passaggio 2b: estrazione delle funzionalità",
|
7 |
+
"step3a:正在训练模型": "Passaggio 3a: è iniziato l'addestramento del modello",
|
8 |
+
"训练结束, 您可查看控制台训练日志或实验文件夹下的train.log": "Addestramento completato. ",
|
9 |
+
"全流程结束!": "Tutti i processi sono stati completati!",
|
10 |
+
"本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责. <br>如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录<b>LICENSE</b>.": "Questo software è open source con licenza MIT. <br>Se non si accetta questa clausola, non è possibile utilizzare o fare riferimento a codici e file all'interno del pacchetto software. <b>Contratto-LICENZA.txt</b> per dettagli.",
|
11 |
+
"模型推理": "Inferenza del modello",
|
12 |
+
"推理音色": "Voce di inferenza:",
|
13 |
+
"刷新音色列表和索引路径": "Aggiorna l'elenco delle voci e il percorso dell'indice",
|
14 |
+
"卸载音色省显存": "Scarica la voce per risparmiare memoria della GPU:",
|
15 |
+
"请选择说话人id": "Seleziona ID locutore/cantante:",
|
16 |
+
"男转女推荐+12key, 女转男推荐-12key, 如果音域爆炸导致音色失真也可以自己调整到合适音域. ": "Tonalità +12 consigliata per la conversione da maschio a femmina e tonalità -12 per la conversione da femmina a maschio. ",
|
17 |
+
"变调(整数, 半音数量, 升八度12降八度-12)": "Trasposizione (numero intero, numero di semitoni, alza di un'ottava: 12, abbassa di un'ottava: -12):",
|
18 |
+
"输入待处理音频文件路径(默认是正确格式示例)": "Immettere il percorso del file audio da elaborare (l'impostazione predefinita è l'esempio di formato corretto):",
|
19 |
+
"选择音高提取算法,输入歌声可用pm提速,harvest低音好但巨慢无比,crepe效果好但吃GPU": "Seleziona l'algoritmo di estrazione del tono (\"pm\": estrazione più veloce ma risultato di qualità inferiore; \"harvest\": bassi migliori ma estremamente lenti; \"crepe\": qualità migliore ma utilizzo intensivo della GPU):",
|
20 |
+
">=3则使用对harvest音高识别的结果使用中值滤波,数值为滤波半径,使用可以削弱哑音": "Se >=3: applica il filtro mediano ai risultati del pitch raccolto. ",
|
21 |
+
"特征检索库文件路径,为空则使用下拉的选择结果": "Percorso del file di indice delle caratteristiche. ",
|
22 |
+
"自动检测index路径,下拉式选择(dropdown)": "Rileva automaticamente il percorso dell'indice e seleziona dal menu a tendina:",
|
23 |
+
"特征文件路径": "Percorso del file delle caratteristiche:",
|
24 |
+
"检索特征占比": "Rapporto funzionalità di ricerca (controlla la forza dell'accento, troppo alto ha artefatti):",
|
25 |
+
"后处理重采样至最终采样率,0为不进行重采样": "Ricampiona l'audio di output in post-elaborazione alla frequenza di campionamento finale. ",
|
26 |
+
"输入源音量包络替换输出音量包络融合比例,越靠近1越使用输出包络": "Regola il ridimensionamento dell'inviluppo del volume. ",
|
27 |
+
"保护清辅音和呼吸声,防止电音撕裂等artifact,拉满0.5不开启,调低加大保护力度但可能降低索引效果": "Proteggi le consonanti senza voce e i suoni del respiro per evitare artefatti come il tearing nella musica elettronica. ",
|
28 |
+
"F0曲线文件, 可选, 一行一个音高, 代替默认F0及升降调": "File curva F0 (opzionale). ",
|
29 |
+
"转换": "Convertire",
|
30 |
+
"输出信息": "Informazioni sull'uscita",
|
31 |
+
"输出音频(右下角三个点,点了可以下载)": "Esporta audio (clicca sui tre puntini in basso a destra per scaricarlo)",
|
32 |
+
"批量转换, 输入待转换音频文件夹, 或上传多个音频文件, 在指定文件夹(默认opt)下输出转换的音频. ": "Conversione massiva. Inserisci il percorso della cartella che contiene i file da convertire o carica più file audio. I file convertiti finiranno nella cartella specificata. (default: opt) ",
|
33 |
+
"指定输出文件夹": "Specifica la cartella di output:",
|
34 |
+
"输入待处理音频文件夹路径(去文件管理器地址栏拷就行了)": "Immettere il percorso della cartella audio da elaborare (copiarlo dalla barra degli indirizzi del file manager):",
|
35 |
+
"也可批量输入音频文件, 二选一, 优先读文件夹": "Puoi anche inserire file audio in massa. ",
|
36 |
+
"导出文件格式": "Formato file di esportazione",
|
37 |
+
"伴奏人声分离&去混响&去回声": "Separazione voce/accompagnamento",
|
38 |
+
"人声伴奏分离批量处理, 使用UVR5模型。 <br>合格的文件夹路径格式举例: E:\\codes\\py39\\vits_vc_gpu\\白鹭霜华测试样例(去文件管理器地址栏拷就行了)。 <br>模型分为三类: <br>1、保留人声:不带和声的音频选这个,对主人声保留比HP5更好。内置HP2和HP3两个模型,HP3可能轻微漏伴奏但对主人声保留比HP2稍微好一丁点; <br>2、仅保留主人声:带和声的音频选这个,对主人声可能有削弱。内置HP5一个模型; <br> 3、去混响、去延迟模型(by FoxJoy):<br> (1)MDX-Net(onnx_dereverb):对于双通道混响是最好的选择,不能去除单通道混响;<br> (234)DeEcho:去除延迟效果。Aggressive比Normal去除得更彻底,DeReverb额外去除混响,可去除单声道混响,但是对高频重的板式混响去不干净。<br>去混响/去延迟,附:<br>1、DeEcho-DeReverb模型的耗时是另外2个DeEcho模型的接近2倍;<br>2、MDX-Net-Dereverb模型挺慢的;<br>3、个人推荐的最干净的配置是先MDX-Net再DeEcho-Aggressive。": "Elaborazione batch per la separazione dell'accompagnamento vocale utilizzando il modello UVR5.<br>Esempio di un formato di percorso di cartella valido: D:\\path\\to\\input\\folder (copialo dalla barra degli indirizzi del file manager).<br>Il modello è suddiviso in tre categorie:<br>1. Conserva la voce: scegli questa opzione per l'audio senza armonie. <br>2. Mantieni solo la voce principale: scegli questa opzione per l'audio con armonie. <br>3. Modelli di de-riverbero e de-delay (di FoxJoy):<br> (1) MDX-Net: la scelta migliore per la rimozione del riverbero stereo ma non può rimuovere il riverbero mono;<br><br>Note di de-riverbero/de-delay:<br>1. Il tempo di elaborazione per il modello DeEcho-DeReverb è circa il doppio rispetto agli altri due modelli DeEcho.<br>2. Il modello MDX-Net-Dereverb è piuttosto lento.<br>3. La configurazione più pulita consigliata consiste nell'applicare prima MDX-Net e poi DeEcho-Aggressive.",
|
39 |
+
"输入待处理音频文件夹路径": "Immettere il percorso della cartella audio da elaborare:",
|
40 |
+
"模型": "Modello",
|
41 |
+
"指定输出主人声文件夹": "Specifica la cartella di output per le voci:",
|
42 |
+
"指定输出非主人声文件夹": "Specificare la cartella di output per l'accompagnamento:",
|
43 |
+
"训练": "Addestramento",
|
44 |
+
"step1: 填写实验配置. 实验数据放在logs下, 每个实验一个文件夹, 需手工输入实验名路径, 内含实验配置, 日志, 训练得到的模型文件. ": "Passaggio 1: compilare la configurazione sperimentale. ",
|
45 |
+
"输入实验名": "Inserisci il nome dell'esperimento:",
|
46 |
+
"目标采样率": "Frequenza di campionamento target:",
|
47 |
+
"模型是否带音高指导(唱歌一定要, 语音可以不要)": "Se il modello ha una guida del tono (necessario per il canto, facoltativo per il parlato):",
|
48 |
+
"版本": "Versione",
|
49 |
+
"提取音高和处理数据使用的CPU进程数": "Numero di processi CPU utilizzati per l'estrazione del tono e l'elaborazione dei dati:",
|
50 |
+
"step2a: 自动遍历训练文件夹下所有可解码成音频的文件并进行切片归一化, 在实验目录下生成2个wav文件夹; 暂时只支持单人训练. ": "Passaggio 2a: attraversa automaticamente tutti i file nella cartella di addestramento che possono essere decodificati in audio ed esegui la normalizzazione delle sezioni. ",
|
51 |
+
"输入训练文件夹路径": "Inserisci il percorso della cartella di addestramento:",
|
52 |
+
"请指定说话人id": "Si prega di specificare l'ID del locutore/cantante:",
|
53 |
+
"处理数据": "Processa dati",
|
54 |
+
"step2b: 使用CPU提取音高(如果模型带音高), 使用GPU提取特征(选择卡号)": "Passaggio 2b: utilizzare la CPU per estrarre il tono (se il modello ha il tono), utilizzare la GPU per estrarre le caratteristiche (selezionare l'indice GPU):",
|
55 |
+
"以-分隔输入使用的卡号, 例如 0-1-2 使用卡0和卡1和卡2": "Inserisci gli indici GPU separati da '-', ad esempio 0-1-2 per utilizzare GPU 0, 1 e 2:",
|
56 |
+
"显卡信息": "Informazioni GPU",
|
57 |
+
"选择音高提取算法:输入歌声可用pm提速,高质量语音但CPU差可用dio提速,harvest质量更好但慢": "Seleziona l'algoritmo di estrazione del tono (\"pm\": estrazione più rapida ma parlato di qualità inferiore; \"dio\": parlato migliorato ma estrazione più lenta; \"harvest\": migliore qualità ma estrazione più lenta):",
|
58 |
+
"特征提取": "Estrazione delle caratteristiche",
|
59 |
+
"step3: 填写训练设置, 开始训练模型和索引": "Passaggio 3: compilare le impostazioni di addestramento e avviare l'addestramento del modello e dell'indice",
|
60 |
+
"保存频率save_every_epoch": "Frequenza di salvataggio (save_every_epoch):",
|
61 |
+
"总训练轮数total_epoch": "Epoch totali di addestramento (total_epoch):",
|
62 |
+
"每张显卡的batch_size": "Dimensione batch per GPU:",
|
63 |
+
"是否仅保存最新的ckpt文件以节省硬盘空间": "Salva solo l'ultimo file '.ckpt' per risparmiare spazio su disco:",
|
64 |
+
"否": "NO",
|
65 |
+
"是否缓存所有训练集至显存. 10min以下小数据可缓存以加速训练, 大数据缓存会炸显存也加不了多少速": "Memorizza nella cache tutti i set di addestramento nella memoria della GPU. ",
|
66 |
+
"是否在每次保存时间点将最终小模型保存至weights文件夹": "Salva un piccolo modello finale nella cartella \"weights\" in ogni punto di salvataggio:",
|
67 |
+
"加载预训练底模G路径": "Carica il percorso G del modello base pre-addestrato:",
|
68 |
+
"加载预训练底模D路径": "Carica il percorso D del modello base pre-addestrato:",
|
69 |
+
"训练模型": "Addestra modello",
|
70 |
+
"训练特征索引": "Addestra indice delle caratteristiche",
|
71 |
+
"一键训练": "Addestramento con un clic",
|
72 |
+
"ckpt处理": "Elaborazione ckpt",
|
73 |
+
"模型融合, 可用于测试音色融合": "Model fusion, può essere utilizzato per testare la fusione timbrica",
|
74 |
+
"A模型路径": "Percorso per il modello A:",
|
75 |
+
"B模型路径": "Percorso per il modello B:",
|
76 |
+
"A模型权重": "Peso (w) per il modello A:",
|
77 |
+
"模型是否带音高指导": "Se il modello ha una guida del tono:",
|
78 |
+
"要置入的模型信息": "Informazioni sul modello da posizionare:",
|
79 |
+
"保存的模型名不带后缀": "Nome del modello salvato (senza estensione):",
|
80 |
+
"模型版本型号": "Versione dell'architettura del modello:",
|
81 |
+
"融合": "Fusione",
|
82 |
+
"修改模型信息(仅支持weights文件夹下提取的小模型文件)": "Modifica le informazioni sul modello (supportato solo per i file di modello di piccole dimensioni estratti dalla cartella 'weights')",
|
83 |
+
"模型路径": "Percorso al modello:",
|
84 |
+
"要改的模型信息": "Informazioni sul modello da modificare:",
|
85 |
+
"保存的文件名, 默认空为和源文件同名": "Salva il nome del file (predefinito: uguale al file di origine):",
|
86 |
+
"修改": "Modificare",
|
87 |
+
"查看模型信息(仅支持weights文件夹下提取的小模型文件)": "Visualizza le informazioni sul modello (supportato solo per file di modello piccoli estratti dalla cartella 'weights')",
|
88 |
+
"查看": "Visualizzazione",
|
89 |
+
"模型提取(输入logs文件夹下大文件模型路径),适用于训一半不想训了模型没有自动提取保存小文件模型,或者想测试中间模型的情况": "Estrazione del modello (inserire il percorso del modello di file di grandi dimensioni nella cartella \"logs\"). ",
|
90 |
+
"保存名": "Salva nome:",
|
91 |
+
"模型是否带音高指导,1是0否": "Se il modello ha una guida del tono (1: sì, 0: no):",
|
92 |
+
"提取": "Estrai",
|
93 |
+
"Onnx导出": "Esporta Onnx",
|
94 |
+
"RVC模型路径": "Percorso modello RVC:",
|
95 |
+
"Onnx输出路径": "Percorso di esportazione Onnx:",
|
96 |
+
"导出Onnx模型": "Esporta modello Onnx",
|
97 |
+
"常见问题解答": "FAQ (Domande frequenti)",
|
98 |
+
"招募音高曲线前端编辑器": "Reclutamento di redattori front-end per curve di tono",
|
99 |
+
"加开发群联系我xxxxx": "Unisciti al gruppo di sviluppo e contattami a xxxxx",
|
100 |
+
"点击查看交流、问题反馈群号": "Fare clic per visualizzare il numero del gruppo di comunicazione e feedback sui problemi",
|
101 |
+
"xxxxx": "xxxxx",
|
102 |
+
"加载模型": "Carica modello",
|
103 |
+
"Hubert模型": "Modello Hubert",
|
104 |
+
"选择.pth文件": "Seleziona il file .pth",
|
105 |
+
"选择.index文件": "Seleziona il file .index",
|
106 |
+
"选择.npy文件": "Seleziona il file .npy",
|
107 |
+
"输入设备": "Dispositivo di input",
|
108 |
+
"输出设备": "Dispositivo di uscita",
|
109 |
+
"音频设备(请使用同种类驱动)": "Dispositivo audio (utilizzare lo stesso tipo di driver)",
|
110 |
+
"响应阈值": "Soglia di risposta",
|
111 |
+
"音调设置": "Impostazioni del tono",
|
112 |
+
"Index Rate": "Tasso di indice",
|
113 |
+
"常规设置": "Impostazioni generali",
|
114 |
+
"采样长度": "Lunghezza del campione",
|
115 |
+
"淡入淡出长度": "Lunghezza dissolvenza",
|
116 |
+
"额外推理时长": "Tempo di inferenza extra",
|
117 |
+
"输入降噪": "Riduzione del rumore in ingresso",
|
118 |
+
"输出降噪": "Riduzione del rumore in uscita",
|
119 |
+
"性能设置": "Impostazioni delle prestazioni",
|
120 |
+
"开始音频转换": "Avvia la conversione audio",
|
121 |
+
"停止音频转换": "Arresta la conversione audio",
|
122 |
+
"推理时间(ms):": "Tempo di inferenza (ms):",
|
123 |
+
"请选择pth文件": "请选择pth 文件",
|
124 |
+
"请选择index文件": "请选择index文件",
|
125 |
+
"hubert模型路径不可包含中文": "hubert 模型路径不可包含中文",
|
126 |
+
"pth文件路径不可包含中文": "pth è un'app per il futuro",
|
127 |
+
"index文件路径不可包含中文": "index文件路径不可包含中文",
|
128 |
+
"音高算法": "音高算法",
|
129 |
+
"harvest进程数": "harvest进程数"
|
130 |
+
}
|
i18n/es_ES.json
ADDED
@@ -0,0 +1,132 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"很遗憾您这没有能用的显卡来支持您训练": "Lamentablemente, no tiene una tarjeta gráfica adecuada para soportar su entrenamiento",
|
3 |
+
"是": "Sí",
|
4 |
+
"step1:正在处理数据": "Paso 1: Procesando datos",
|
5 |
+
"step2a:无需提取音高": "Paso 2a: No es necesario extraer el tono",
|
6 |
+
"step2b:正在提取特征": "Paso 2b: Extrayendo características",
|
7 |
+
"step3a:正在训练模型": "Paso 3a: Entrenando el modelo",
|
8 |
+
"训练结束, 您可查看控制台训练日志或实验文件夹下的train.log": "Entrenamiento finalizado, puede ver el registro de entrenamiento en la consola o en el archivo train.log en la carpeta del experimento",
|
9 |
+
"全流程结束!": "¡Todo el proceso ha terminado!",
|
10 |
+
"本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责. <br>如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录<b>LICENSE</b>.": "Este software es de código abierto bajo la licencia MIT, el autor no tiene ningún control sobre el software, y aquellos que usan el software y difunden los sonidos exportados por el software son los únicos responsables.<br>Si no está de acuerdo con esta cláusula , no puede utilizar ni citar ningún código ni archivo del paquete de software Consulte el directorio raíz <b>Agreement-LICENSE.txt</b> para obtener más información.",
|
11 |
+
"模型推理": "inferencia del modelo",
|
12 |
+
"推理音色": "inferencia de voz",
|
13 |
+
"刷新音色列表和索引路径": "Actualizar la lista de timbres e índice de rutas",
|
14 |
+
"卸载音色省显存": "Descargue la voz para ahorrar memoria GPU",
|
15 |
+
"请选择说话人id": "seleccione una identificación de altavoz",
|
16 |
+
"男转女推荐+12key, 女转男推荐-12key, 如果音域爆炸导致音色失真也可以自己调整到合适音域. ": "Tecla +12 recomendada para conversión de voz de hombre a mujer, tecla -12 para conversión de voz de mujer a hombre. Si el rango de tono es demasiado amplio y causa distorsión, ajústelo usted mismo a un rango adecuado.",
|
17 |
+
"变调(整数, 半音数量, 升八度12降八度-12)": "Cambio de tono (entero, número de semitonos, subir una octava +12 o bajar una octava -12)",
|
18 |
+
"输入待处理音频文件路径(默认是正确格式示例)": "Ingrese la ruta del archivo del audio que se procesará (el formato predeterminado es el ejemplo correcto)",
|
19 |
+
"选择音高提取算法,输入歌声可用pm提速,harvest低音好但巨慢无比,crepe效果好但吃GPU": "Seleccione el algoritmo para la extracción de tono. Use 'pm' para acelerar las voces cantadas, o use 'harvest' para mejorar las voces bajas, pero es extremadamente lento.",
|
20 |
+
"crepe_hop_length": "Crepe Hop Length (Only applies to crepe): Hop length refers to the time it takes for the speaker to jump to a dramatic pitch. Lower hop lengths take more time to infer but are more pitch accurate.",
|
21 |
+
"特征检索库文件路径": "Ruta del archivo de la base de datos de búsqueda de características",
|
22 |
+
">=3则使用对harvest音高识别的结果使用中值滤波,数值为滤波半径,使用可以削弱哑音": ">=3则使用对harvest音高识别的结果使用中值滤波,数值为滤波半径,使用可以削弱哑音",
|
23 |
+
"特征检索库文件路径,为空则使用下拉的选择结果": "特征检索库文件路径,为空则使用下拉的选择结果",
|
24 |
+
"自动检测index路径,下拉式选择(dropdown)": "自动检测index路径,下拉式选择(dropdown)",
|
25 |
+
"特征文件路径": "Ruta del archivo de características",
|
26 |
+
"检索特征占比": "Proporción de función de búsqueda",
|
27 |
+
"后处理重采样至最终采样率,0为不进行重采样": "Remuestreo posterior al proceso a la tasa de muestreo final, 0 significa no remuestrear",
|
28 |
+
"输入源音量包络替换输出音量包络融合比例,越靠近1越使用输出包络": "Proporción de fusión para reemplazar el sobre de volumen de entrada con el sobre de volumen de salida, cuanto más cerca de 1, más se utiliza el sobre de salida",
|
29 |
+
"保护清辅音和呼吸声,防止电音撕裂等artifact,拉满0.5不开启,调低加大保护力度但可能降低索引效果": "Proteger las consonantes claras y la respiración, prevenir artefactos como la distorsión de sonido electrónico, 0.5 no está activado, reducir aumentará la protección pero puede reducir el efecto del índice",
|
30 |
+
"F0曲线文件, 可选, 一行一个音高, 代替默认F0及升降调": "Archivo de curva F0, opcional, un tono por línea, en lugar de F0 predeterminado y cambio de tono",
|
31 |
+
"转换": "Conversión",
|
32 |
+
"输出信息": "Información de salida",
|
33 |
+
"输出音频(右下角三个点,点了可以下载)": "Salida de audio (haga clic en los tres puntos en la esquina inferior derecha para descargar)",
|
34 |
+
"批量转换, 输入待转换音频文件夹, 或上传多个音频文件, 在指定文件夹(默认opt)下输出转换的音频. ": "Conversión por lotes, ingrese la carpeta que contiene los archivos de audio para convertir o cargue varios archivos de audio. El audio convertido se emitirá en la carpeta especificada (opción predeterminada).",
|
35 |
+
"指定输出文件夹": "Especificar carpeta de salida",
|
36 |
+
"输入待处理音频文件夹路径(去文件管理器地址栏拷就行了)": "Ingrese la ruta a la carpeta de audio que se procesará (simplemente cópiela desde la barra de direcciones del administrador de archivos)",
|
37 |
+
"也可批量输入音频文件, 二选一, 优先读文件夹": "También se pueden ingresar múltiples archivos de audio, cualquiera de las dos opciones, con prioridad dada a la carpeta",
|
38 |
+
"导出文件格式": "Formato de archivo de exportación",
|
39 |
+
"伴奏人声分离&去混响&去回声": "Separación de voz acompañante & eliminación de reverberación & eco",
|
40 |
+
"人声伴奏分离批量处理, 使用UVR5模型。 <br>合格的文件夹路径格式举例: E:\\codes\\py39\\vits_vc_gpu\\白鹭霜华测试样例(去文件管理器地址栏拷就行了)。 <br>模型分为三类: <br>1、保留人声:不带和声的音频选这个,对主人声保留比HP5更好。内置HP2和HP3两个模型,HP3可能轻微漏伴奏但对主人声保留比HP2稍微好一丁点; <br>2、仅保留主人声:带和声的音频选这个,对主人声可能有削弱。内置HP5一个模型; <br> 3、去混响、去延迟模型(by FoxJoy):<br> (1)MDX-Net(onnx_dereverb):对于双通道混响是最好的选择,不能去除单通道混响;<br> (234)DeEcho:去除延迟效果。Aggressive比Normal去除得更彻底,DeReverb额外去除混响,可去除单声道混响,但是对高频重的板式混响去不干净。<br>去混响/去延迟,附:<br>1、DeEcho-DeReverb模型的耗时是另外2个DeEcho模型的接近2倍;<br>2、MDX-Net-Dereverb模型挺慢的;<br>3、个人推荐的最干净的配置是先MDX-Net再DeEcho-Aggressive。": "Procesamiento por lotes para la separación de acompañamiento vocal utilizando el modelo UVR5.<br>Ejemplo de formato de ruta de carpeta válido: D:\\ruta\\a\\la\\carpeta\\de\\entrada (copiar desde la barra de direcciones del administrador de archivos).<br>El modelo se divide en tres categorías:<br>1. Preservar voces: Elija esta opción para audio sin armonías. Preserva las voces mejor que HP5. Incluye dos modelos incorporados: HP2 y HP3. HP3 puede filtrar ligeramente el acompañamiento pero conserva las voces un poco mejor que HP2.<br>2. Preservar solo voces principales: Elija esta opción para audio con armonías. Puede debilitar las voces principales. Incluye un modelo incorporado: HP5.<br>3. Modelos de des-reverberación y des-retardo (por FoxJoy):<br> (1) MDX-Net: La mejor opción para la eliminación de reverberación estéreo pero no puede eliminar la reverberación mono;<br> (234) DeEcho: Elimina efectos de retardo. El modo Agresivo elimina más a fondo que el modo Normal. DeReverb adicionalmente elimina la reverberación y puede eliminar la reverberación mono, pero no muy efectivamente para contenido de alta frecuencia fuertemente reverberado.<br>Notas de des-reverberación/des-retardo:<br>1. El tiempo de procesamiento para el modelo DeEcho-DeReverb es aproximadamente el doble que los otros dos modelos DeEcho.<br>2. El modelo MDX-Net-Dereverb es bastante lento.<br>3. La configuración más limpia recomendada es aplicar primero MDX-Net y luego DeEcho-Agresivo.",
|
41 |
+
"输入待处理音频文件夹路径": "Ingrese la ruta a la carpeta de audio que se procesará",
|
42 |
+
"模型": "Modelo",
|
43 |
+
"指定输出主人声文件夹": "Especifique la carpeta de salida para la voz principal",
|
44 |
+
"指定输出非主人声文件夹": "Especifique la carpeta de salida para las voces no principales",
|
45 |
+
"训练": "Entrenamiento",
|
46 |
+
"step1: 填写实验配置. 实验数据放在logs下, 每个实验一个文件夹, 需手工输入实验名路径, 内含实验配置, 日志, 训练得到的模型文件. ": "paso 1: Complete la configuración del experimento. Los datos del experimento se almacenan en el directorio 'logs', con cada experimento en una carpeta separada. La ruta del nombre del experimento debe ingresarse manualmente y debe contener la configuración del experimento, los registros y los archivos del modelo entrenado.",
|
47 |
+
"输入实验名": "Ingrese el nombre del modelo",
|
48 |
+
"目标采样率": "Tasa de muestreo objetivo",
|
49 |
+
"模型是否带音高指导(唱歌一定要, 语音可以不要)": "Si el modelo tiene guía de tono (necesaria para cantar, pero no para hablar)",
|
50 |
+
"版本": "Versión",
|
51 |
+
"提取音高和处理数据使用的CPU进程数": "Número de procesos de CPU utilizados para extraer el tono y procesar los datos",
|
52 |
+
"step2a: 自动遍历训练文件夹下所有可解码成音频的文件并进行切片归一化, 在实验目录下生成2个wav文件夹; 暂时只支持单人训练. ": "paso 2a: recorra automáticamente la carpeta de capacitación y corte y normalice todos los archivos de audio que se pueden decodificar en audio. Se generarán dos carpetas 'wav' en el directorio del experimento. Actualmente, solo se admite la capacitación de una sola persona.",
|
53 |
+
"输入训练文件夹路径": "Introduzca la ruta de la carpeta de entrenamiento",
|
54 |
+
"请指定说话人id": "Especifique el ID del hablante",
|
55 |
+
"处理数据": "Procesar datos",
|
56 |
+
"step2b: 使用CPU提取音高(如果模型带音高), 使用GPU提取特征(选择卡号)": "paso 2b: use la CPU para extraer el tono (si el modelo tiene guía de tono) y la GPU para extraer características (seleccione el número de tarjeta).",
|
57 |
+
"以-分隔输入使用的卡号, 例如 0-1-2 使用卡0和卡1和卡2": "Separe los números de identificación de la GPU con '-' al ingresarlos. Por ejemplo, '0-1-2' significa usar GPU 0, GPU 1 y GPU 2.",
|
58 |
+
"显卡信息": "información de la GPU",
|
59 |
+
"选择音高提取算法:输入歌声可用pm提速,高质量语音但CPU差可用dio提速,harvest质量更好但慢": "Seleccione el algoritmo de extracción de tono: utilice 'pm' para un procesamiento más rápido de la voz cantada, 'dio' para un discurso de alta calidad pero un procesamiento más lento y 'cosecha' para obtener la mejor calidad pero un procesamiento más lento.",
|
60 |
+
"特征提取": "Extracción de características",
|
61 |
+
"step3: 填写训练设置, 开始训练模型和索引": "Paso 3: complete la configuración de entrenamiento y comience a entrenar el modelo y el índice.",
|
62 |
+
"保存频率save_every_epoch": "Frecuencia de guardado (save_every_epoch)",
|
63 |
+
"总训练轮数total_epoch": "Total de épocas de entrenamiento (total_epoch)",
|
64 |
+
"每张显卡的batch_size": "Tamaño del lote (batch_size) por tarjeta gráfica",
|
65 |
+
"是否仅保存最新的ckpt文件以节省硬盘空间": "Si guardar solo el archivo ckpt más reciente para ahorrar espacio en disco",
|
66 |
+
"否": "No",
|
67 |
+
"是否缓存所有训练集至显存. 10min以下小数据可缓存以加速训练, 大数据缓存会炸显存也加不了多少速": "Si almacenar en caché todos los conjuntos de entrenamiento en la memoria de la GPU. Los conjuntos de datos pequeños (menos de 10 minutos) se pueden almacenar en caché para acelerar el entrenamiento, pero el almacenamiento en caché de conjuntos de datos grandes puede causar errores de memoria en la GPU y no aumenta la velocidad de manera significativa.",
|
68 |
+
"是否在每次保存时间点将最终小模型保存至weights文件夹": "¿Guardar el pequeño modelo final en la carpeta 'weights' en cada punto de guardado?",
|
69 |
+
"加载预训练底模G路径": "Cargue la ruta G del modelo base preentrenada.",
|
70 |
+
"加载预训练底模D路径": "Cargue la ruta del modelo D base preentrenada.",
|
71 |
+
"训练模型": "Entrenar Modelo",
|
72 |
+
"训练特征索引": "Índice de características del Entrenamiento",
|
73 |
+
"一键训练": "One-click training. (Not working on this fork)",
|
74 |
+
"ckpt处理": "Procesamiento de recibos",
|
75 |
+
"模型融合, 可用于测试音色融合": "Fusión de modelos, se puede utilizar para fusionar diferentes voces",
|
76 |
+
"A模型路径": "Modelo A ruta.",
|
77 |
+
"B模型路径": "Modelo B ruta.",
|
78 |
+
"A模型权重": "Un peso modelo para el modelo A.",
|
79 |
+
"模型是否带音高指导": "Si el modelo tiene guía de tono.",
|
80 |
+
"要置入的模型信息": "Información del modelo a colocar.",
|
81 |
+
"保存的模型名不带后缀": "Nombre del modelo guardado sin extensión.",
|
82 |
+
"模型版本型号": "Versión y modelo del modelo",
|
83 |
+
"融合": "Fusión.",
|
84 |
+
"修改模型信息(仅支持weights文件夹下提取的小模型文件)": "Modificar la información del modelo (solo admite archivos de modelos pequeños extraídos en la carpeta de pesos).",
|
85 |
+
"模型路径": "Ruta del modelo",
|
86 |
+
"要改的模型信息": "Información del modelo a modificar",
|
87 |
+
"保存的文件名, 默认空为和源文件同名": "Nombre del archivo que se guardará, el valor predeterminado es el mismo que el nombre del archivo de origen",
|
88 |
+
"修改": "Modificar",
|
89 |
+
"查看模型信息(仅支持weights文件夹下提取的小模型文件)": "Ver información del modelo (solo aplicable a archivos de modelos pequeños extraídos de la carpeta 'pesos')",
|
90 |
+
"查看": "Ver",
|
91 |
+
"模型提取(输入logs文件夹下大文件模型路径),适用于训一半不想训了模型没有自动提取保存小文件模型,或者想测试中间模型的情况": "Extracción de modelo (ingrese la ruta de un archivo de modelo grande en la carpeta 'logs'), aplicable cuando desea extraer un archivo de modelo pequeño después de entrenar a mitad de camino y no se guardó automáticamente, o cuando desea probar un modelo intermedio",
|
92 |
+
"保存名": "Guardar nombre",
|
93 |
+
"模型是否带音高指导,1是0否": "Si el modelo tiene guía de tono, 1 para sí, 0 para no",
|
94 |
+
"提取": "Extracter",
|
95 |
+
"Onnx导出": "Exportar Onnx",
|
96 |
+
"RVC模型路径": "Ruta del modelo RVC",
|
97 |
+
"Onnx输出路径": "Ruta de salida Onnx",
|
98 |
+
"导出Onnx模型": "Exportar modelo Onnx",
|
99 |
+
"常见问题解答": "Preguntas frecuentes",
|
100 |
+
"招募音高曲线前端编辑器": "Reclutar editores front-end para curvas de tono",
|
101 |
+
"加开发群联系我xxxxx": "Únase al grupo de desarrollo para contactarme en xxxxx",
|
102 |
+
"点击查看交流、问题反馈群号": "Haga clic para ver el número de grupo de comunicación y comentarios sobre problemas",
|
103 |
+
"xxxxx": "xxxxx",
|
104 |
+
"加载模型": "Cargar modelo",
|
105 |
+
"Hubert模型": "Modelo de Hubert ",
|
106 |
+
"选择.pth文件": "Seleccionar archivo .pth",
|
107 |
+
"选择.index文件": "Select .index file",
|
108 |
+
"选择.npy文件": "Seleccionar archivo .npy",
|
109 |
+
"输入设备": "Dispositivo de entrada",
|
110 |
+
"输出设备": "Dispositivo de salida",
|
111 |
+
"音频设备(请使用同种类驱动)": "Dispositivo de audio (utilice el mismo tipo de controlador)",
|
112 |
+
"响应阈值": "Umbral de respuesta",
|
113 |
+
"音调设置": "Ajuste de tono",
|
114 |
+
"Index Rate": "Tasa de índice",
|
115 |
+
"常规设置": "Configuración general",
|
116 |
+
"采样长度": "Longitud de muestreo",
|
117 |
+
"淡入淡出长度": "Duración del fundido de entrada/salida",
|
118 |
+
"额外推理时长": "Tiempo de inferencia adicional",
|
119 |
+
"输入降噪": "Reducción de ruido de entrada",
|
120 |
+
"输出降噪": "Reducción de ruido de salida",
|
121 |
+
"性能设置": "Configuración de rendimiento",
|
122 |
+
"开始音频转换": "Iniciar conversión de audio",
|
123 |
+
"停止音频转换": "Detener la conversión de audio",
|
124 |
+
"推理时间(ms):": "Inferir tiempo (ms):",
|
125 |
+
"请选择pth文件": "请选择pth文件",
|
126 |
+
"请选择index文件": "请选择index文件",
|
127 |
+
"hubert模型路径不可包含中文": "hubert模型路径不可包含中文",
|
128 |
+
"pth文件路径不可包含中文": "pth文件路径不可包含中文",
|
129 |
+
"index文件路径不可包含中文": "index文件路径不可包含中文",
|
130 |
+
"音高算法": "音高算法",
|
131 |
+
"harvest进程数": "harvest进程数"
|
132 |
+
}
|
i18n/it_IT.json
ADDED
@@ -0,0 +1,130 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"很遗憾您这没有能用的显卡来支持您训练": "Sfortunatamente, non è disponibile alcuna GPU compatibile per supportare l'addestramento.",
|
3 |
+
"是": "SÌ",
|
4 |
+
"step1:正在处理数据": "Passaggio 1: elaborazione dei dati",
|
5 |
+
"step2a:无需提取音高": "Step 2a: Saltare l'estrazione del tono",
|
6 |
+
"step2b:正在提取特征": "Passaggio 2b: estrazione delle funzionalità",
|
7 |
+
"step3a:正在训练模型": "Passaggio 3a: è iniziato l'addestramento del modello",
|
8 |
+
"训练结束, 您可查看控制台训练日志或实验文件夹下的train.log": "Addestramento completato. ",
|
9 |
+
"全流程结束!": "Tutti i processi sono stati completati!",
|
10 |
+
"本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责. <br>如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录<b>LICENSE</b>.": "Questo software è open source con licenza MIT. <br>Se non si accetta questa clausola, non è possibile utilizzare o fare riferimento a codici e file all'interno del pacchetto software. <b>Contratto-LICENZA.txt</b> per dettagli.",
|
11 |
+
"模型推理": "Inferenza del modello",
|
12 |
+
"推理音色": "Voce di inferenza:",
|
13 |
+
"刷新音色列表和索引路径": "Aggiorna l'elenco delle voci e il percorso dell'indice",
|
14 |
+
"卸载音色省显存": "Scarica la voce per risparmiare memoria della GPU:",
|
15 |
+
"请选择说话人id": "Seleziona ID locutore/cantante:",
|
16 |
+
"男转女推荐+12key, 女转男推荐-12key, 如果音域爆炸导致音色失真也可以自己调整到合适音域. ": "Tonalità +12 consigliata per la conversione da maschio a femmina e tonalità -12 per la conversione da femmina a maschio. ",
|
17 |
+
"变调(整数, 半音数量, 升八度12降八度-12)": "Trasposizione (numero intero, numero di semitoni, alza di un'ottava: 12, abbassa di un'ottava: -12):",
|
18 |
+
"输入待处理音频文件路径(默认是正确格式示例)": "Immettere il percorso del file audio da elaborare (l'impostazione predefinita è l'esempio di formato corretto):",
|
19 |
+
"选择音高提取算法,输入歌声可用pm提速,harvest低音好但巨慢无比,crepe效果好但吃GPU": "Seleziona l'algoritmo di estrazione del tono (\"pm\": estrazione più veloce ma risultato di qualità inferiore; \"harvest\": bassi migliori ma estremamente lenti; \"crepe\": qualità migliore ma utilizzo intensivo della GPU):",
|
20 |
+
">=3则使用对harvest音高识别的结果使用中值滤波,数值为滤波半径,使用可以削弱哑音": "Se >=3: applica il filtro mediano ai risultati del pitch raccolto. ",
|
21 |
+
"特征检索库文件路径,为空则使用下拉的选择结果": "Percorso del file di indice delle caratteristiche. ",
|
22 |
+
"自动检测index路径,下拉式选择(dropdown)": "Rileva automaticamente il percorso dell'indice e seleziona dal menu a tendina:",
|
23 |
+
"特征文件路径": "Percorso del file delle caratteristiche:",
|
24 |
+
"检索特征占比": "Rapporto funzionalità di ricerca (controlla la forza dell'accento, troppo alto ha artefatti):",
|
25 |
+
"后处理重采样至最终采样率,0为不进行重采样": "Ricampiona l'audio di output in post-elaborazione alla frequenza di campionamento finale. ",
|
26 |
+
"输入源音量包络替换输出音量包络融合比例,越靠近1越使用输出包络": "Regola il ridimensionamento dell'inviluppo del volume. ",
|
27 |
+
"保护清辅音和呼吸声,防止电音撕裂等artifact,拉满0.5不开启,调低加大保护力度但可能降低索引效果": "Proteggi le consonanti senza voce e i suoni del respiro per evitare artefatti come il tearing nella musica elettronica. ",
|
28 |
+
"F0曲线文件, 可选, 一行一个音高, 代替默认F0及升降调": "File curva F0 (opzionale). ",
|
29 |
+
"转换": "Convertire",
|
30 |
+
"输出信息": "Informazioni sull'uscita",
|
31 |
+
"输出音频(右下角三个点,点了可以下载)": "Esporta audio (clicca sui tre puntini in basso a destra per scaricarlo)",
|
32 |
+
"批量转换, 输入待转换音频文件夹, 或上传多个音频文件, 在指定文件夹(默认opt)下输出转换的音频. ": "Conversione massiva. Inserisci il percorso della cartella che contiene i file da convertire o carica più file audio. I file convertiti finiranno nella cartella specificata. (default: opt) ",
|
33 |
+
"指定输出文件夹": "Specifica la cartella di output:",
|
34 |
+
"输入待处理音频文件夹路径(去文件管理器地址栏拷就行了)": "Immettere il percorso della cartella audio da elaborare (copiarlo dalla barra degli indirizzi del file manager):",
|
35 |
+
"也可批量输入音频文件, 二选一, 优先读文件夹": "Puoi anche inserire file audio in massa. ",
|
36 |
+
"导出文件格式": "Formato file di esportazione",
|
37 |
+
"伴奏人声分离&去混响&去回声": "Separazione voce/accompagnamento",
|
38 |
+
"人声伴奏分离批量处理, 使用UVR5模型。 <br>合格的文件夹路径格式举例: E:\\codes\\py39\\vits_vc_gpu\\白鹭霜华测试样例(去文件管理器地址栏拷就行了)。 <br>模型分为三类: <br>1、保留人声:不带和声的音频选这个,对主人声保留比HP5更好。内置HP2和HP3两个模型,HP3可能轻微漏伴奏但对主人声保留比HP2稍微好一丁点; <br>2、仅保留主人声:带和声的音频选这个,对主人声可能有削弱。内置HP5一个模型; <br> 3、去混响、去延迟模型(by FoxJoy):<br> (1)MDX-Net(onnx_dereverb):对于双通道混响是最好的选择,不能去除单通道混响;<br> (234)DeEcho:去除延迟效果。Aggressive比Normal去除得更彻底,DeReverb额外去除混响,可去除单声道混响,但是对高频重的板式混响去不干净。<br>去混响/去延迟,附:<br>1、DeEcho-DeReverb模型的耗时是另外2个DeEcho模型的接近2倍;<br>2、MDX-Net-Dereverb模型挺慢的;<br>3、个人推荐的最干净的配置是先MDX-Net再DeEcho-Aggressive。": "Elaborazione batch per la separazione dell'accompagnamento vocale utilizzando il modello UVR5.<br>Esempio di un formato di percorso di cartella valido: D:\\path\\to\\input\\folder (copialo dalla barra degli indirizzi del file manager).<br>Il modello è suddiviso in tre categorie:<br>1. Conserva la voce: scegli questa opzione per l'audio senza armonie. <br>2. Mantieni solo la voce principale: scegli questa opzione per l'audio con armonie. <br>3. Modelli di de-riverbero e de-delay (di FoxJoy):<br> (1) MDX-Net: la scelta migliore per la rimozione del riverbero stereo ma non può rimuovere il riverbero mono;<br><br>Note di de-riverbero/de-delay:<br>1. Il tempo di elaborazione per il modello DeEcho-DeReverb è circa il doppio rispetto agli altri due modelli DeEcho.<br>2. Il modello MDX-Net-Dereverb è piuttosto lento.<br>3. La configurazione più pulita consigliata consiste nell'applicare prima MDX-Net e poi DeEcho-Aggressive.",
|
39 |
+
"输入待处理音频文件夹路径": "Immettere il percorso della cartella audio da elaborare:",
|
40 |
+
"模型": "Modello",
|
41 |
+
"指定输出主人声文件夹": "Specifica la cartella di output per le voci:",
|
42 |
+
"指定输出非主人声文件夹": "Specificare la cartella di output per l'accompagnamento:",
|
43 |
+
"训练": "Addestramento",
|
44 |
+
"step1: 填写实验配置. 实验数据放在logs下, 每个实验一个文件夹, 需手工输入实验名路径, 内含实验配置, 日志, 训练得到的模型文件. ": "Passaggio 1: compilare la configurazione sperimentale. ",
|
45 |
+
"输入实验名": "Inserisci il nome dell'esperimento:",
|
46 |
+
"目标采样率": "Frequenza di campionamento target:",
|
47 |
+
"模型是否带音高指导(唱歌一定要, 语音可以不要)": "Se il modello ha una guida del tono (necessario per il canto, facoltativo per il parlato):",
|
48 |
+
"版本": "Versione",
|
49 |
+
"提取音高和处理数据使用的CPU进程数": "Numero di processi CPU utilizzati per l'estrazione del tono e l'elaborazione dei dati:",
|
50 |
+
"step2a: 自动遍历训练文件夹下所有可解码成音频的文件并进行切片归一化, 在实验目录下生成2个wav文件夹; 暂时只支持单人训练. ": "Passaggio 2a: attraversa automaticamente tutti i file nella cartella di addestramento che possono essere decodificati in audio ed esegui la normalizzazione delle sezioni. ",
|
51 |
+
"输入训练文件夹路径": "Inserisci il percorso della cartella di addestramento:",
|
52 |
+
"请指定说话人id": "Si prega di specificare l'ID del locutore/cantante:",
|
53 |
+
"处理数据": "Processa dati",
|
54 |
+
"step2b: 使用CPU提取音高(如果模型带音高), 使用GPU提取特征(选择卡号)": "Passaggio 2b: utilizzare la CPU per estrarre il tono (se il modello ha il tono), utilizzare la GPU per estrarre le caratteristiche (selezionare l'indice GPU):",
|
55 |
+
"以-分隔输入使用的卡号, 例如 0-1-2 使用卡0和卡1和卡2": "Inserisci gli indici GPU separati da '-', ad esempio 0-1-2 per utilizzare GPU 0, 1 e 2:",
|
56 |
+
"显卡信息": "Informazioni GPU",
|
57 |
+
"选择音高提取算法:输入歌声可用pm提速,高质量语音但CPU差可用dio提速,harvest质量更好但慢": "Seleziona l'algoritmo di estrazione del tono (\"pm\": estrazione più rapida ma parlato di qualità inferiore; \"dio\": parlato migliorato ma estrazione più lenta; \"harvest\": migliore qualità ma estrazione più lenta):",
|
58 |
+
"特征提取": "Estrazione delle caratteristiche",
|
59 |
+
"step3: 填写训练设置, 开始训练模型和索引": "Passaggio 3: compilare le impostazioni di addestramento e avviare l'addestramento del modello e dell'indice",
|
60 |
+
"保存频率save_every_epoch": "Frequenza di salvataggio (save_every_epoch):",
|
61 |
+
"总训练轮数total_epoch": "Epoch totali di addestramento (total_epoch):",
|
62 |
+
"每张显卡的batch_size": "Dimensione batch per GPU:",
|
63 |
+
"是否仅保存最新的ckpt文件以节省硬盘空间": "Salva solo l'ultimo file '.ckpt' per risparmiare spazio su disco:",
|
64 |
+
"否": "NO",
|
65 |
+
"是否缓存所有训练集至显存. 10min以下小数据可缓存以加速训练, 大数据缓存会炸显存也加不了多少速": "Memorizza nella cache tutti i set di addestramento nella memoria della GPU. ",
|
66 |
+
"是否在每次保存时间点将最终小模型保存至weights文件夹": "Salva un piccolo modello finale nella cartella \"weights\" in ogni punto di salvataggio:",
|
67 |
+
"加载预训练底模G路径": "Carica il percorso G del modello base pre-addestrato:",
|
68 |
+
"加载预训练底模D路径": "Carica il percorso D del modello base pre-addestrato:",
|
69 |
+
"训练模型": "Addestra modello",
|
70 |
+
"训练特征索引": "Addestra indice delle caratteristiche",
|
71 |
+
"一键训练": "Addestramento con un clic",
|
72 |
+
"ckpt处理": "Elaborazione ckpt",
|
73 |
+
"模型融合, 可用于测试音色融合": "Model fusion, può essere utilizzato per testare la fusione timbrica",
|
74 |
+
"A模型路径": "Percorso per il modello A:",
|
75 |
+
"B模型路径": "Percorso per il modello B:",
|
76 |
+
"A模型权重": "Peso (w) per il modello A:",
|
77 |
+
"模型是否带音高指导": "Se il modello ha una guida del tono:",
|
78 |
+
"要置入的模型信息": "Informazioni sul modello da posizionare:",
|
79 |
+
"保存的模型名不带后缀": "Nome del modello salvato (senza estensione):",
|
80 |
+
"模型版本型号": "Versione dell'architettura del modello:",
|
81 |
+
"融合": "Fusione",
|
82 |
+
"修改模型信息(仅支持weights文件夹下提取的小模型文件)": "Modifica le informazioni sul modello (supportato solo per i file di modello di piccole dimensioni estratti dalla cartella 'weights')",
|
83 |
+
"模型路径": "Percorso al modello:",
|
84 |
+
"要改的模型信息": "Informazioni sul modello da modificare:",
|
85 |
+
"保存的文件名, 默认空为和源文件同名": "Salva il nome del file (predefinito: uguale al file di origine):",
|
86 |
+
"修改": "Modificare",
|
87 |
+
"查看模型信息(仅支持weights文件夹下提取的小模型文件)": "Visualizza le informazioni sul modello (supportato solo per file di modello piccoli estratti dalla cartella 'weights')",
|
88 |
+
"查看": "Visualizzazione",
|
89 |
+
"模型提取(输入logs文件夹下大文件模型路径),适用于训一半不想训了模型没有自动提取保存小文件模型,或者想测试中间模型的情况": "Estrazione del modello (inserire il percorso del modello di file di grandi dimensioni nella cartella \"logs\"). ",
|
90 |
+
"保存名": "Salva nome:",
|
91 |
+
"模型是否带音高指导,1是0否": "Se il modello ha una guida del tono (1: sì, 0: no):",
|
92 |
+
"提取": "Estrai",
|
93 |
+
"Onnx导出": "Esporta Onnx",
|
94 |
+
"RVC模型路径": "Percorso modello RVC:",
|
95 |
+
"Onnx输出路径": "Percorso di esportazione Onnx:",
|
96 |
+
"导出Onnx模型": "Esporta modello Onnx",
|
97 |
+
"常见问题解答": "FAQ (Domande frequenti)",
|
98 |
+
"招募音高曲线前端编辑器": "Reclutamento di redattori front-end per curve di tono",
|
99 |
+
"加开发群联系我xxxxx": "Unisciti al gruppo di sviluppo e contattami a xxxxx",
|
100 |
+
"点击查看交流、问题反馈群号": "Fare clic per visualizzare il numero del gruppo di comunicazione e feedback sui problemi",
|
101 |
+
"xxxxx": "xxxxx",
|
102 |
+
"加载模型": "Carica modello",
|
103 |
+
"Hubert模型": "Modello Hubert",
|
104 |
+
"选择.pth文件": "Seleziona il file .pth",
|
105 |
+
"选择.index文件": "Seleziona il file .index",
|
106 |
+
"选择.npy文件": "Seleziona il file .npy",
|
107 |
+
"输入设备": "Dispositivo di input",
|
108 |
+
"输出设备": "Dispositivo di uscita",
|
109 |
+
"音频设备(请使用同种类驱动)": "Dispositivo audio (utilizzare lo stesso tipo di driver)",
|
110 |
+
"响应阈值": "Soglia di risposta",
|
111 |
+
"音调设置": "Impostazioni del tono",
|
112 |
+
"Index Rate": "Tasso di indice",
|
113 |
+
"常规设置": "Impostazioni generali",
|
114 |
+
"采样长度": "Lunghezza del campione",
|
115 |
+
"淡入淡出长度": "Lunghezza dissolvenza",
|
116 |
+
"额外推理时长": "Tempo di inferenza extra",
|
117 |
+
"输入降噪": "Riduzione del rumore in ingresso",
|
118 |
+
"输出降噪": "Riduzione del rumore in uscita",
|
119 |
+
"性能设置": "Impostazioni delle prestazioni",
|
120 |
+
"开始音频转换": "Avvia la conversione audio",
|
121 |
+
"停止音频转换": "Arresta la conversione audio",
|
122 |
+
"推理时间(ms):": "Tempo di inferenza (ms):",
|
123 |
+
"请选择pth文件": "请选择pth 文件",
|
124 |
+
"请选择index文件": "请选择index文件",
|
125 |
+
"hubert模型路径不可包含中文": "hubert 模型路径不可包含中文",
|
126 |
+
"pth文件路径不可包含中文": "pth è un'app per il futuro",
|
127 |
+
"index文件路径不可包含中文": "index文件路径不可包含中文",
|
128 |
+
"音高算法": "音高算法",
|
129 |
+
"harvest进程数": "harvest进程数"
|
130 |
+
}
|
i18n/ja_JP.json
ADDED
@@ -0,0 +1,126 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"很遗憾您这没有能用的显卡来支持您训练": "トレーニングに対応したGPUが動作しないのは残念です。",
|
3 |
+
"是": "はい",
|
4 |
+
"step1:正在处理数据": "step1:処理中のデータ",
|
5 |
+
"step2a:无需提取音高": "step2a:ピッチの抽出は不要",
|
6 |
+
"step2b:正在提取特征": "step2b:抽出される特徴量",
|
7 |
+
"step3a:正在训练模型": "step3a:トレーニング中のモデル",
|
8 |
+
"训练结束, 您可查看控制台训练日志或实验文件夹下的train.log": "トレーニング終了時に、トレーニングログやフォルダ内のtrain.logを確認することができます",
|
9 |
+
"全流程结束!": "全工程が完了!",
|
10 |
+
"本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责. <br>如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录<b>使用需遵守的协议-LICENSE.txt</b>.": "本ソフトウェアはMITライセンスに基づくオープンソースであり、作者は本ソフトウェアに対していかなる強制力も持ちません。本ソフトウェアの利用者および本ソフトウェアから派生した音源(成果物)を配布する者は、本ソフトウェアに対して自身で責任を負うものとします。 <br>この条項に同意しない場合、パッケージ内のコードやファイルを使用や参照を禁じます。詳しくは<b>使用需遵守的协议-LICENSE.txt</b>をご覧ください.",
|
11 |
+
"模型推理": "モデル推論",
|
12 |
+
"推理音色": "音源推論",
|
13 |
+
"刷新音色列表和索引路径": "音源リストとインデックスパスの更新",
|
14 |
+
"卸载音色省显存": "音源を削除してメモリを節約",
|
15 |
+
"请选择说话人id": "話者IDを選択してください",
|
16 |
+
"男转女推荐+12key, 女转男推荐-12key, 如果音域爆炸导致音色失真也可以自己调整到合适音域. ": "男性から女性へは+12キーをお勧めします。女性から男性へは-12キーをお勧めします。音域が広すぎて音質が劣化した場合は、適切な音域に自分で調整することもできます。",
|
17 |
+
"变调(整数, 半音数量, 升八度12降八度-12)": "ピッチ変更(整数、半音数、上下オクターブ12-12)",
|
18 |
+
"输入待处理音频文件路径(默认是正确格式示例)": "処理対象音声ファイルのパスを入力してください(デフォルトは正しいフォーマットの例です)",
|
19 |
+
"选择音高提取算法,输入歌声可用pm提速,harvest低音好但巨慢无比,crepe效果好但吃GPU": "选择音高提取算法,输入歌声可用pm提速,harvest低音好但巨慢无比,crepe效果好但吃GPU",
|
20 |
+
"crepe_hop_length": "Crepe Hop Length (Only applies to crepe): Hop length refers to the time it takes for the speaker to jump to a dramatic pitch. Lower hop lengths take more time to infer but are more pitch accurate.",
|
21 |
+
"特征检索库文件路径": "特徴量検索データベースのファイルパス",
|
22 |
+
">=3则使用对harvest音高识别的结果使用中值滤波,数值为滤波半径,使用可以削弱哑音": ">=3 次に、harvestピッチの認識結果に対してメディアンフィルタを使用します。値はフィルター半径で、ミュートを減衰させるために使用します。",
|
23 |
+
"特征检索库文件路径,为空则使用下拉的选择结果": "特徴検索ライブラリへのパス 空の場合はドロップダウンで選択",
|
24 |
+
"自动检测index路径,下拉式选择(dropdown)": "インデックスパスの自動検出 ドロップダウンで選択",
|
25 |
+
"特征文件路径": "特徴量ファイルのパス",
|
26 |
+
"检索特征占比": "検索特徴率",
|
27 |
+
"后处理重采样至最终采样率,0为不进行重采样": "最終的なサンプリングレートへのポストプロセッシングのリサンプリング リサンプリングしない場合は0",
|
28 |
+
"输入源音量包络替换输出音量包络融合比例,越靠近1越使用输出包络": "入力ソースの音量エンベロープと出力音量エンベロープの融合率 1に近づくほど、出力音量エンベロープの割合が高くなる",
|
29 |
+
"保护清辅音和呼吸声,防止电音撕裂等artifact,拉满0.5不开启,调低加大保护力度但可能降低索引效果": "保护清辅音和呼吸声,防止电音撕裂等artifact,拉满0.5不开启,调低加大保护力度但可能降低索引效果",
|
30 |
+
"F0曲线文件, 可选, 一行一个音高, 代替默认F0及升降调": "F0(最低共振周波数)カーブファイル(オプション、1行に1ピッチ、デフォルトのF0(最低共振周波数)とエレベーションを置き換えます。)",
|
31 |
+
"转换": "変換",
|
32 |
+
"输出信息": "出力情報",
|
33 |
+
"输出音频(右下角三个点,点了可以下载)": "出力音声(右下の三点をクリックしてダウンロードできます)",
|
34 |
+
"批量转��, 输入待转换音频文件夹, 或上传多个音频文件, 在指定文件夹(默认opt)下输出转换的音频. ": "一括変換、変換する音声フォルダを入力、または複数の音声ファイルをアップロードし、指定したフォルダ(デフォルトのopt)に変換した音声を出力します。",
|
35 |
+
"指定输出文件夹": "出力フォルダを指定してください",
|
36 |
+
"输入待处理音频文件夹路径(去文件管理器地址栏拷就行了)": "処理対象音声フォルダーのパスを入力してください(ファイルマネージャのアドレスバーからコピーしてください)",
|
37 |
+
"也可批量输入音频文件, 二选一, 优先读文件夹": "複数の音声ファイルを一括で入力することもできますが、フォルダーを優先して読み込みます",
|
38 |
+
"导出文件格式": "导出文件格式",
|
39 |
+
"伴奏人声分离&去混响&去回声": "伴奏人声分离&去混响&去回声",
|
40 |
+
"输入待处理音频文件夹路径": "処理するオーディオファイルのフォルダパスを入力してください",
|
41 |
+
"模型": "モデル",
|
42 |
+
"指定输出主人声文件夹": "指定输出主人声文件夹",
|
43 |
+
"指定输出非主人声文件夹": "指定输出非主人声文件夹",
|
44 |
+
"训练": "トレーニング",
|
45 |
+
"step1: 填写实验配置. 实验数据放在logs下, 每个实验一个文件夹, 需手工输入实验名路径, 内含实验配置, 日志, 训练得到的模型文件. ": "ステップ1:実験設定を入力します。実験データはlogsに保存され、各実験にはフォルダーがあります。実験名のパスを手動で入力する必要があり、実験設定、ログ、トレーニングされたモデルファイルが含まれます。",
|
46 |
+
"输入实验名": "モデル名",
|
47 |
+
"目标采样率": "目標サンプリングレート",
|
48 |
+
"模型是否带音高指导(唱歌一定要, 语音可以不要)": "モデルに音高ガイドがあるかどうか(歌唱には必要ですが、音声には必要ありません)",
|
49 |
+
"版本": "バージョン",
|
50 |
+
"提取音高和处理数据使用的CPU进程数": "ピッチの抽出やデータ処理に使用するCPUスレッド数",
|
51 |
+
"step2a: 自动遍历训练文件夹下所有可解码成音频的文件并进行切片归一化, 在实验目录下生成2个wav文件夹; 暂时只支持单人训练. ": "ステップ2a: 訓練フォルダー内のすべての音声ファイルを自動的に探索し、スライスと正規化を行い、2つのwavフォルダーを実験ディレクトリに生成します。現在は一人でのトレーニングのみをサポートしています。",
|
52 |
+
"输入训练文件夹路径": "トレーニング用フォルダのパスを入力してください",
|
53 |
+
"请指定说话人id": "話者IDを指定してください",
|
54 |
+
"处理数据": "データ処理",
|
55 |
+
"step2b: 使用CPU提取音高(如果模型带音高), 使用GPU提取特征(选择卡号)": "ステップ2b: CPUを使用して音高を抽出する(モデルに音高がある場合)、GPUを使用して特徴を抽出する(GPUの番号を選択する)",
|
56 |
+
"以-分隔输入使用的卡号, 例如 0-1-2 使用卡0和卡1和卡2": "ハイフンで区切って使用するGPUの番号を入力します。例えば0-1-2はGPU0、GPU1、GPU2を使用します",
|
57 |
+
"显卡信息": "GPU情報",
|
58 |
+
"选择音高提取算法:输入歌声可用pm提速,高质量语音但CPU差可用dio提速,harvest质量更好但慢": "音高抽出アルゴリズムの選択:歌声を入力する場合は、pmを使用して速度を上げることができます。CPUが低い場合はdioを使用して速度を上げることができます。harvestは品質が高く、精度が高いですが、遅いです。",
|
59 |
+
"特征提取": "特徴抽出",
|
60 |
+
"step3: 填写训练设置, 开始训练模型和索引": "ステップ3: トレーニング設定を入力して、モデルとインデックスのトレーニングを開始します",
|
61 |
+
"保存频率save_every_epoch": "エポックごとの保存頻度",
|
62 |
+
"总训练轮数total_epoch": "総エポック数",
|
63 |
+
"每张显卡的batch_size": "GPUごとのバッチサイズ",
|
64 |
+
"是否仅保存最新的ckpt文件以节省硬盘空间": "ハードディスク容量を節約するため、最新のckptファイルのみを保存するかどうか",
|
65 |
+
"否": "いいえ",
|
66 |
+
"是否缓存所有训练集至显存. 10min以下小数据可缓存以加速训练, 大数据缓存会炸显存也加不了多少速": "すべてのトレーニングデータをメモリにキャッシュするかどうか。10分以下の小さなデータはキャッシュしてトレーニングを高速化できますが、大きなデータをキャッシュするとメモリが破裂し、あまり速度が上がりません。",
|
67 |
+
"是否在每次保存时间点将最终小模型保存至weights文件夹": "各保存時点の小モデルを全部weightsフォルダに保存するか���うか",
|
68 |
+
"加载预训练底模G路径": "事前学習済みのGモデルのパス",
|
69 |
+
"加载预训练底模D路径": "事前学習済みのDモデルのパス",
|
70 |
+
"训练模型": "モデルのトレーニング",
|
71 |
+
"训练特征索引": "特徴インデックスのトレーニング",
|
72 |
+
"一键训练": "One-click training. (Not working on this fork)",
|
73 |
+
"ckpt处理": "ckptファイルの処理",
|
74 |
+
"模型融合, 可用于测试音色融合": "モデルのマージ、音源のマージテストに使用できます",
|
75 |
+
"A模型路径": "Aモデルのパス",
|
76 |
+
"B模型路径": "Bモデルのパス",
|
77 |
+
"A模型权重": "Aモデルの重み",
|
78 |
+
"模型是否带音高指导": "モデルに音高ガイドを付けるかどうか",
|
79 |
+
"要置入的模型信息": "挿入するモデル情報",
|
80 |
+
"保存的模型名不带后缀": "拡張子のない保存するモデル名",
|
81 |
+
"模型版本型号": "モデルのバージョン",
|
82 |
+
"融合": "フュージョン",
|
83 |
+
"修改模型信息(仅支持weights文件夹下提取的小模型文件)": "モデル情報の修正(weightsフォルダから抽出された小さなモデルファイルのみ対応)",
|
84 |
+
"模型路径": "モデルパス",
|
85 |
+
"要改的模型信息": "変更するモデル情報",
|
86 |
+
"保存的文件名, 默认空为和源文件同名": "保存するファイル名、デフォルトでは空欄で元のファイル名と同じ名前になります",
|
87 |
+
"修改": "変更",
|
88 |
+
"查看模型信息(仅支持weights文件夹下提取的小模型文件)": "モデル情報を表示する(小さいモデルファイルはweightsフォルダーからのみサポートされています)",
|
89 |
+
"查看": "表示",
|
90 |
+
"模型提取(输入logs文件夹下大文件模型路径),适用于训一半不想训了模型没有自动提取保存小文件模型,或者想测试中间模型的情况": "モデル抽出(ログフォルダー内の大きなファイルのモデルパスを入力)、モデルを半分までトレーニングし、自動的に小さいファイルモデルを保存しなかったり、中間モデルをテストしたい場合に適用されます。",
|
91 |
+
"保存名": "保存ファイル名",
|
92 |
+
"模型是否带音高指导,1是0否": "モデルに音高ガイドを付けるかどうか、1は付ける、0は付けない",
|
93 |
+
"提取": "抽出",
|
94 |
+
"Onnx导出": "Onnx",
|
95 |
+
"RVC模型路径": "RVCモデルパス",
|
96 |
+
"Onnx输出路径": "Onnx出力パス",
|
97 |
+
"MoeVS模型": "MoeSS?",
|
98 |
+
"导出Onnx模型": "Onnxに変換",
|
99 |
+
"常见问题解答": "よくある質問",
|
100 |
+
"招募音高曲线前端编辑器": "音高曲線フロントエンドエディターを募集",
|
101 |
+
"加开发群联系我xxxxx": "開発グループに参加して私に連絡してくださいxxxxx",
|
102 |
+
"点击查看交流、问题反馈群号": "クリックして交流、問題フィードバックグループ番号を表示",
|
103 |
+
"xxxxx": "xxxxx",
|
104 |
+
"加载模型": "モデルをロード",
|
105 |
+
"Hubert模型": "Hubertモデル",
|
106 |
+
"选择.pth文件": ".pthファイルを選択",
|
107 |
+
"选择.index文件": ".indexファイルを選択",
|
108 |
+
"选择.npy文件": ".npyファイルを選択",
|
109 |
+
"输入设备": "入力デバイス",
|
110 |
+
"输出设备": "出力デバイス",
|
111 |
+
"音频设备(请使用同种类驱动)": "オーディオデバイス(同じ種類のドライバーを使用してください)",
|
112 |
+
"响应阈值": "反応閾値",
|
113 |
+
"音调设置": "音程設定",
|
114 |
+
"Index Rate": "Index Rate",
|
115 |
+
"常规设置": "一般設定",
|
116 |
+
"采样长度": "サンプル長",
|
117 |
+
"淡入淡出长度": "フェードイン/フェードアウト長",
|
118 |
+
"额外推理时长": "追加推論時間",
|
119 |
+
"输入降噪": "入力ノイズの低減",
|
120 |
+
"输出降噪": "出力ノイズの低減",
|
121 |
+
"性能设置": "パフォーマンス設定",
|
122 |
+
"开始音频转换": "音声変換を開始",
|
123 |
+
"停止音频转换": "音声変換を停止",
|
124 |
+
"推理时间(ms):": "推論時間(ms):",
|
125 |
+
"人声伴奏分离批量处理, 使用UVR5模型。 <br>合格的文件夹路径格式举例: E:\\codes\\py39\\vits_vc_gpu\\白鹭霜华测试样例(去文件管理器地址栏拷就行了)。 <br>模型分为三类: <br>1、保留人声:不带和声的音频选这个,对主人声保留比HP5更好。内置HP2和HP3两个模型,HP3可能轻微漏伴奏但对主人声保留比HP2稍微好一丁点; <br>2、仅保留主人声:带和声的音频选这个,对主人声可能有削弱。内置HP5一个模型; <br> 3、去混响、去延迟模型(by FoxJoy):<br> (1)MDX-Net(onnx_dereverb):对于双通道混响是最好的选择,不能去除单通道混响;<br> (234)DeEcho:去除延迟效果。Aggressive比Normal去除得更彻底,DeReverb额外去除混响,可去除单声道混响,但是对高频重的板式混响去不干净。<br>去混响/去延迟,附:<br>1、DeEcho-DeReverb模型的耗时是另外2个DeEcho模型的接近2倍;<br>2、MDX-Net-Dereverb模型挺慢的;<br>3、个人推荐的最干净的配置是先MDX-Net再DeEcho-Aggressive。":"UVR5モデルを使用したボーカル伴奏の分離バッチ処理。<br>有効なフォルダーパスフォーマットの例: D:\\path\\to\\input\\folder (ファイルマネージャのアドレスバーからコピーします)。<br>モデルは三つのカテゴリに分かれています:<br>1. ボーカルを保持: ハーモニーのないオーディオに対してこれを選択します。HP5よりもボーカルをより良く保持します。HP2とHP3の二つの内蔵モデルが含まれています。HP3は伴奏をわずかに漏らす可能性がありますが、HP2よりもわずかにボーカルをより良く保持します。<br>2. 主なボーカルのみを保持: ハーモニーのあるオーディオに対してこれを選択します。主なボーカルを弱める可能性があります。HP5の一つの内蔵モデルが含まれています。<br>3. ディリバーブとディレイモデル (by FoxJoy):<br> (1) MDX-Net: ステレオリバーブの除去に最適な選択肢ですが、モノリバーブは除去できません;<br> (234) DeEcho: ディレイ効果を除去します。AggressiveモードはNormalモードよりも徹底的に除去します。DeReverbはさらにリバーブを除去し、モノリバーブを除去することができますが、高周波のリバーブが強い内容に対しては非常に効果的ではありません。<br>ディリバーブ/ディレイに関する注意点:<br>1. DeEcho-DeReverbモデルの処理時間は、他の二つのDeEchoモデルの約二倍です。<br>2. MDX-Net-Dereverbモデルは非常に遅いです。<br>3. 推奨される最もクリーンな設定は、最初にMDX-Netを適用し、その後にDeEcho-Aggressiveを適用することです。"
|
126 |
+
}
|
i18n/locale_diff.py
ADDED
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import json
|
2 |
+
import os
|
3 |
+
from collections import OrderedDict
|
4 |
+
|
5 |
+
# Define the standard file name
|
6 |
+
standard_file = "zh_CN.json"
|
7 |
+
|
8 |
+
# Find all JSON files in the directory
|
9 |
+
dir_path = "./"
|
10 |
+
languages = [
|
11 |
+
f for f in os.listdir(dir_path) if f.endswith(".json") and f != standard_file
|
12 |
+
]
|
13 |
+
|
14 |
+
# Load the standard file
|
15 |
+
with open(standard_file, "r", encoding="utf-8") as f:
|
16 |
+
standard_data = json.load(f, object_pairs_hook=OrderedDict)
|
17 |
+
|
18 |
+
# Loop through each language file
|
19 |
+
for lang_file in languages:
|
20 |
+
# Load the language file
|
21 |
+
with open(lang_file, "r", encoding="utf-8") as f:
|
22 |
+
lang_data = json.load(f, object_pairs_hook=OrderedDict)
|
23 |
+
|
24 |
+
# Find the difference between the language file and the standard file
|
25 |
+
diff = set(standard_data.keys()) - set(lang_data.keys())
|
26 |
+
|
27 |
+
miss = set(lang_data.keys()) - set(standard_data.keys())
|
28 |
+
|
29 |
+
# Add any missing keys to the language file
|
30 |
+
for key in diff:
|
31 |
+
lang_data[key] = key
|
32 |
+
|
33 |
+
# Del any extra keys to the language file
|
34 |
+
for key in miss:
|
35 |
+
del lang_data[key]
|
36 |
+
|
37 |
+
# Sort the keys of the language file to match the order of the standard file
|
38 |
+
lang_data = OrderedDict(
|
39 |
+
sorted(lang_data.items(), key=lambda x: list(standard_data.keys()).index(x[0]))
|
40 |
+
)
|
41 |
+
|
42 |
+
# Save the updated language file
|
43 |
+
with open(lang_file, "w", encoding="utf-8") as f:
|
44 |
+
json.dump(lang_data, f, ensure_ascii=False, indent=4)
|
45 |
+
f.write("\n")
|
i18n/ru-RU.json
ADDED
@@ -0,0 +1,130 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"很遗憾您这没有能用的显卡来支持您训练": "К сожалению у вас нету видеокарты, которая поддерживает тренировку модели.",
|
3 |
+
"是": "Да",
|
4 |
+
"step1:正在处理数据": "Шаг 1: Переработка данных",
|
5 |
+
"step2a:无需提取音高": "Шаг 2а: Пропуск вытаскивания тональности",
|
6 |
+
"step2b:正在提取特征": "Шаг 2б: Вытаскивание черт",
|
7 |
+
"step3a:正在训练模型": "Шаг 3а: Тренировка модели начата",
|
8 |
+
"训练结束, 您可查看控制台训练日志或实验文件夹下的train.log": "Тренировка завершена. Вы можете проверить логи тренировки в консоли или в файле 'train.log' в папке модели.",
|
9 |
+
"全流程结束!": "Все процессы завершены!",
|
10 |
+
"本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责. <br>如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录<b>LICENSE</b>.": "本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责. <br>如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录<b>LICENSE</b>.",
|
11 |
+
"模型推理": "Обработка модели",
|
12 |
+
"推理音色": "Обработка голоса:",
|
13 |
+
"刷新音色列表和索引路径": "Обновить список голосов и индексов",
|
14 |
+
"卸载音色省显存": "Выгрузить голос для сохранения памяти видеокарты:",
|
15 |
+
"请选择说话人id": "Выбери айди голоса:",
|
16 |
+
"男转女推荐+12key, 女转男推荐-12key, 如果音域爆炸导致音色失真也可以自己调整到合适音域. ": "Рекомендованно +12 для конвертирования мужского голоса в женский и -12 для конвертирования женского в мужской. Если диапазон голоса слищком велик и голос искажается, значение можно изменить на свой вкус.",
|
17 |
+
"变调(整数, 半音数量, 升八度12降八度-12)": "Высота голоса (число, полутоны, поднять на октаву: 12, понизить на октаву: -12):",
|
18 |
+
"输入待处理音频文件路径(默认是正确格式示例)": "Введите путь к аудиофайлу, который хотите переработать (по умолчанию введён правильный формат):",
|
19 |
+
"选择音高提取算法,输入歌声可用pm提速,harvest低音好但巨慢无比,crepe效果好但吃GPU": "Выберите алгоритм вытаскивания тональности ('pm': быстрое извлечение но качество речи хуже; 'harvest': бассы лучше но очень медленный; 'crepe': лучшее качество но сильно использует видеокарту):",
|
20 |
+
">=3则使用对harvest音高识别的结果使用中值滤波,数值为滤波半径,使用可以削弱哑音": "Если больше 3: применить медианную фильтрацию к вытащенным тональностям. Значение контролирует радиус фильтра и может уменьшить излишнее дыхание.",
|
21 |
+
"特征检索库文件路径,为空则使用下拉的选择结果": "Путь к файлу индекса черт. Оставьте пустым, чтобы использовать выбранный результат из списка:",
|
22 |
+
"自动检测index路径,下拉式选择(dropdown)": "Автоматически найти путь к индексу и выбрать его из списка:",
|
23 |
+
"特征文件路径": "Путь к файлу черт:",
|
24 |
+
"检索特征占比": "Соотношение поиска черт:",
|
25 |
+
"后处理重采样至最终采样率,0为不进行重采样": "Изменить частоту дискретизации в выходном файле на финальную. Поставьте 0, чтобы ничего не изменялось:",
|
26 |
+
"输入源音量包络替换输出音量包络融合比例,越靠近1越使用输出包络": "Использовать громкость входного файла для замены или перемешивания с громкостью выходного файла. Чем ближе соотношение к 1, тем больше используется звука из выходного файла:",
|
27 |
+
"保护清辅音和呼吸声,防止电音撕裂等artifact,拉满0.5不开启,调低加大保护力度但可能降低索引效果": "Защитить глухие со��ласные и звуки дыхания для предотвращения артефактов, например разрывание в электронной музыке. Поставьте на 0.5, чтобы выключить. Уменьшите значение для повышения защиты, но при этом может ухудшиться аккуратность индексирования:",
|
28 |
+
"F0曲线文件, 可选, 一行一个音高, 代替默认F0及升降调": "Файл дуги F0 (не обязательно). Одна тональность на каждую строчку. Заменяет обычный F0 и модуляцию тональности:",
|
29 |
+
"转换": "Конвертировать",
|
30 |
+
"输出信息": "Выходная информация",
|
31 |
+
"输出音频(右下角三个点,点了可以下载)": "Экспортировать аудиофайл (нажми на три точки в правом нижнем углу для загрузки)",
|
32 |
+
"批量转换, 输入待转换音频文件夹, 或上传多个音频文件, 在指定文件夹(默认opt)下输出转换的音频. ": "Конвертировать пачкой. Введите путь к папке, в которой находятся файлы для конвертирования или выложите несколько аудиофайлов. Сконвертированные файлы будут сохранены в указанной папке (по умолчанию 'opt').",
|
33 |
+
"指定输出文件夹": "Укажите выходную папку:",
|
34 |
+
"输入待处理音频文件夹路径(去文件管理器地址栏拷就行了)": "Введите путь к папке с аудио для переработки:",
|
35 |
+
"也可批量输入音频文件, 二选一, 优先读文件夹": "Вы также можете выложить аудиофайлы пачкой. Выберите одно из двух. Приоритет отдаётся считыванию из папки.",
|
36 |
+
"导出文件格式": "Формат выходного файла",
|
37 |
+
"伴奏人声分离&去混响&去回声": "Отделение вокала/инструментала и убирание эхо",
|
38 |
+
"人声伴奏分离批量处理, 使用UVR5模型。 <br>合格的文件夹路径格式举例: E:\\codes\\py39\\vits_vc_gpu\\白鹭霜华测试样例(去文件管理器地址栏拷就行了)。 <br>模型分为三类: <br>1、保留人声:不带和声的音频选这个,对主人声保留比HP5更好。内置HP2和HP3两个模型,HP3可能轻微漏伴奏但对主人声保留比HP2稍微好一丁点; <br>2、仅保留主人声:带和声的音频选这个,对主人声可能有削弱。内置HP5一个模型; <br> 3、去混响、去延迟模型(by FoxJoy):<br> (1)MDX-Net(onnx_dereverb):对于双通道混响是最好的选择,不能去除单通道混响;<br> (234)DeEcho:去除延迟效果。Aggressive比Normal去除得更彻底,DeReverb额外去除混响,可去除单声道混响,但是对高频重的板式混响去不干净。<br>去混响/去延迟,附:<br>1、DeEcho-DeReverb模型的耗时是另外2个DeEcho模型的接近2倍;<br>2、MDX-Net-Dereverb模型挺慢的;<br>3、个人推荐的最干净的配置是先MDX-Net再DeEcho-Aggressive。": "Пакетная обработка для разделения вокального сопровождения с использованием модели UVR5.<br>Пример допустимого формата пути к папке: D:\\path\\to\\input\\folder<br> Модель разделена на три категории:<br>1. Сохранить вокал: выберите этот вариант для звука без гармоний. Он сохраняет вокал лучше, чем HP5. Он включает в себя две встроенные модели: HP2 и HP3. HP3 может немного пропускать инструментал, но сохраняет вокал немного лучше, чем HP2.<br>2. Сохранить только основной вокал: выберите этот вариант для звука с гармониями. Это может ослабить основной вокал. Он включает одну встроенную модель: HP5.<br>3. Модели удаления реверберации и задержки (от FoxJoy):<br> (1) MDX-Net: лучший выбор для удаления стереореверберации, но он не может удалить монореверберацию;<br> (234) DeEcho: удаляет эффекты задержки. Агрессивный режим удаляет более тщательно, чем Нормальный режим. DeReverb дополнительно удаляет реверберацию и может удалять монореверберацию, но не очень эффективно для сильно ревербери��ованного высокочастотного контента.<br>Примечания по удалению реверберации/задержки:<br>1. Время обработки для модели DeEcho-DeReverb примерно в два раза больше, чем для двух других моделей DeEcho.<br>2. Модель MDX-Net-Dereverb довольно медленная.<br>3. Рекомендуемая самая чистая конфигурация — сначала применить MDX-Net, а затем DeEcho-Aggressive.",
|
39 |
+
"输入待处理音频文件夹路径": "Введите путь к папке с аудиофайлами для переработки:",
|
40 |
+
"模型": "Модели",
|
41 |
+
"指定输出主人声文件夹": "Введите путь к папке для вокала:",
|
42 |
+
"指定输出非主人声文件夹": "Введите путь к папке для инструментала:",
|
43 |
+
"训练": "Тренировка",
|
44 |
+
"step1: 填写实验配置. 实验数据放在logs下, 每个实验一个文件夹, 需手工输入实验名路径, 内含实验配置, 日志, 训练得到的模型文件. ": "Шаг 1: Заполните настройки модели. Данные модели сохранены в папку 'logs' и для каждой модели создаётся отдельная папка. Введите вручную путь к настройкам для модели, в которой находятся логи и тренировочные файлы.",
|
45 |
+
"输入实验名": "Введите название модели:",
|
46 |
+
"目标采样率": "Частота дискретизации модели:",
|
47 |
+
"模型是否带音高指导(唱歌一定要, 语音可以不要)": "Наведение по тональности у модели (обязательно для пения, необязательно для речи):",
|
48 |
+
"版本": "Версия",
|
49 |
+
"提取音高和处理数据使用的CPU进程数": "Число процессов ЦП, используемое для вытаскивания тональностей и обрабротки данных:",
|
50 |
+
"step2a: 自动遍历训练文件夹下所有可解码成音频的文件并进行切片归一化, 在实验目录下生成2个wav文件夹; 暂时只支持单人训练. ": "Шаг 2а: Автоматически пройтись по всем аудиофайлам в папке тренировки и нормализировать куски. Создаст 2 папки wav в папке модели. В данных момент поддерживается тренировка только одного голоса.",
|
51 |
+
"输入训练文件夹路径": "Введите путь к папке тренировки:",
|
52 |
+
"请指定说话人id": "Введите айди голоса:",
|
53 |
+
"处理数据": "Переработать данные",
|
54 |
+
"step2b: 使用CPU提取音高(如果模型带音高), 使用GPU提取特征(选择卡号)": "Шаг 2б: Вытащить тональности с помошью процессора (если в модели есть тональности), вытащить черты с помощью видеокарты (выберите какой):",
|
55 |
+
"以-分隔输入使用的卡号, 例如 0-1-2 使用卡0和卡1和卡2": "Введите, какие(-ую) видеокарты(-у) хотите использовать через '-', например 0-1-2, чтобы использовать видеокарту 0, 1 и 2:",
|
56 |
+
"显卡信息": "Информация о видеокартах",
|
57 |
+
"选择音高提取算法:输入歌声可用pm提速,高质量语音但CPU差可用dio提速,harvest质量更好但慢": "Выберите алгоритм вытаскивания тональности ('pm': быстрое извлечение но качество речи хуже; 'harvest': бассы лучше но очень медленный; 'crepe': лучшее качество но сильно использует видеокарту):",
|
58 |
+
"特征提取": "Вытаскивание черт",
|
59 |
+
"step3: 填写训练设置, 开始训练模型和索引": "Шаг 3: Заполните остальные настройки тренировки и начните тренировать модель и индекс",
|
60 |
+
"保存频率save_every_epoch": "Частота сохранения (save_every_epoch):",
|
61 |
+
"总训练轮数total_epoch": "Полное количество эпох (total_epoch):",
|
62 |
+
"每张显卡的batch_size": "Размер пачки для видеокарты:",
|
63 |
+
"是否仅保存最新的ckpt文件以节省硬盘空间": "Сохранять только последний файл '.ckpt', чтобы сохранить место на диске:",
|
64 |
+
"否": "Нет",
|
65 |
+
"是否缓存所有训练集至显存. 10min以下小数据可缓存以加速训练, 大数据缓存会炸显存也加不了多少速": "Кэшировать все тренировочные сеты в видеопамять. Кэширование маленький датасетов (меньше 10 минут) может ускорить тренировку, но кэширование больших, наоборот, займёт много видеопамяти и не сильно ускорит тренировку:",
|
66 |
+
"是否在每次保存时间点将最终小模型保存至weights文件夹": "Сохранять маленькую финальную модель в папку 'weights' на каждой точке сохранения:",
|
67 |
+
"加载预训练底模G路径": "Путь к натренированой базовой модели G:",
|
68 |
+
"加载预训练底模D路径": "Путь к натренированой базовой модели D:",
|
69 |
+
"训练模型": "Тренировать модель",
|
70 |
+
"训练特征索引": "Тренировать индекс черт",
|
71 |
+
"一键训练": "Тренировка одним нажатием",
|
72 |
+
"ckpt处理": "Обработка ckpt",
|
73 |
+
"模型融合, 可用于测试音色融合": "Слияние моделей, может быть использовано для проверки слияния тембра",
|
74 |
+
"A模型路径": "Путь к модели А:",
|
75 |
+
"B模型路径": "Путь к модели Б:",
|
76 |
+
"A模型权重": "Вес (w) модели А::",
|
77 |
+
"模型是否带音高指导": "Есть ли у модели наведение по тональности (1: да, 0: нет):",
|
78 |
+
"要置入的模型信息": "Информация о модели:",
|
79 |
+
"保存的模型名不带后缀": "Название сохранённой модели (без расширения):",
|
80 |
+
"模型版本型号": "Версия архитектуры модели:",
|
81 |
+
"融合": "Слияние",
|
82 |
+
"修改模型信息(仅支持weights文件夹下提取的小模型文件)": "Модифицировать информацию о модели (поддерживается только для маленких моделей, взятых из папки 'weights')",
|
83 |
+
"模型路径": "Путь к папке:",
|
84 |
+
"要改的模型信息": "Информация о модели, которую нужно модифицировать:",
|
85 |
+
"保存的文件名, 默认空为和源文件同名": "Название сохранённого файла (по умолчанию такое же, как и входного):",
|
86 |
+
"修改": "Модифицировать",
|
87 |
+
"查看模型信息(仅支持weights文件夹下提取的小模型文件)": "Просмотреть информацию о модели (поддерживается только для маленких моделей, взятых из папки 'weights')",
|
88 |
+
"查看": "Просмотр",
|
89 |
+
"模型提取(输入logs文件夹下大文件模型路径),适用于训一半不想训了模型没有自动提取保存小文件模型,或者想测试中间模型的情况": "Вытаскивание модели (введите путь к большому файлу модели в папке 'logs'). Полезно, если Вам нужно заверщить тренировку и вручную достать и сохранить маленький файл модели, или если Вам нужно проверить незаконченную модель:",
|
90 |
+
"保存名": "Имя сохранённого файла:",
|
91 |
+
"模型是否带音高指导,1是0否": "Есть ли у модели наведение по тональности (1: да, 0: нет):",
|
92 |
+
"提取": "Вытащить",
|
93 |
+
"Onnx导出": "Экспортировать Onnx",
|
94 |
+
"RVC模型路径": "Путь к модели RVC:",
|
95 |
+
"Onnx输出路径": "Путь для экспотрированного Onnx:",
|
96 |
+
"导出Onnx模型": "Экспортировать Onnx модель",
|
97 |
+
"常见问题解答": "ЧаВО (Часто задаваемые вопросы)",
|
98 |
+
"招募音高曲线前端编辑器": "Использование фронтенд редакторов для тональных дуг",
|
99 |
+
"加开发群联系我xxxxx": "Присоединитесь к группе разработки и свяжитесь со мной по xxxxx",
|
100 |
+
"点击查看交流、问题反馈群号": "Нажмите, чтобы просмотреть номер группы коммуникации и отзывах о проблемах",
|
101 |
+
"xxxxx": "xxxxx",
|
102 |
+
"加载模型": "Загрузить модель",
|
103 |
+
"Hubert模型": "Модель Hubert",
|
104 |
+
"选择.pth文件": "Выбрать файл .pth",
|
105 |
+
"选择.index文件": "Выбрать файл .index",
|
106 |
+
"选择.npy文件": "Выбрать файл .npy",
|
107 |
+
"输入设备": "Входное устройство",
|
108 |
+
"输出设备": "Выходное устройство",
|
109 |
+
"音频设备(请使用同种类驱动)": "Аудио устройство (пожалуйста используйте такой=же тип драйвера)",
|
110 |
+
"响应阈值": "Порог ответа",
|
111 |
+
"音调设置": "Настройки тональности",
|
112 |
+
"Index Rate": "Темп индекса",
|
113 |
+
"常规设置": "Основные настройки",
|
114 |
+
"采样长度": "Длина сэмпла",
|
115 |
+
"淡入淡出长度": "Длина затухания",
|
116 |
+
"额外推理时长": "Доп. время переработки",
|
117 |
+
"输入降噪": "Уменьшения шума во входной информации",
|
118 |
+
"输出降噪": "Уменьшения шума во выходной информации",
|
119 |
+
"性能设置": "Настройки быстроты",
|
120 |
+
"开始音频转换": "Начать конвертацию аудио",
|
121 |
+
"停止音频转换": "Закончить конвертацию аудио",
|
122 |
+
"推理时间(ms):": "Время переработки (мс):",
|
123 |
+
"请选择pth文件": "请选择pth文件",
|
124 |
+
"请选择index文件": "请选择index文件",
|
125 |
+
"hubert模型路径不可包含中文": "hubert模型路径不可包含中文",
|
126 |
+
"pth文件路径不可包含中文": "pth文件路径不可包含中文",
|
127 |
+
"index文件路径不可包含中文": "index文件路径不可包含中文",
|
128 |
+
"音高算法": "音高算法",
|
129 |
+
"harvest进程数": "harvest进程数"
|
130 |
+
}
|
i18n/tr_TR.json
ADDED
@@ -0,0 +1,130 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"很遗憾您这没有能用的显卡来支持您训练": "Maalesef, eğitiminizi desteklemek için uyumlu bir GPU bulunmamaktadır.",
|
3 |
+
"是": "Evet",
|
4 |
+
"step1:正在处理数据": "Adım 1: Veri işleme",
|
5 |
+
"step2a:无需提取音高": "Adım 2a: Pitch çıkartma adımını atlama",
|
6 |
+
"step2b:正在提取特征": "Adım 2b: Özelliklerin çıkarılması",
|
7 |
+
"step3a:正在训练模型": "Adım 3a: Model eğitimi başladı",
|
8 |
+
"训练结束, 您可查看控制台训练日志或实验文件夹下的train.log": "Eğitim tamamlandı. Eğitim günlüklerini konsolda veya deney klasörü altındaki train.log dosyasında kontrol edebilirsiniz.",
|
9 |
+
"全流程结束!": "Tüm işlemler tamamlandı!",
|
10 |
+
"本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责. <br>如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录<b>LICENSE</b>.": "Bu yazılım, MIT lisansı altında açık kaynaklıdır. Yazarın yazılım üzerinde herhangi bir kontrolü yoktur. Yazılımı kullanan ve yazılım tarafından dışa aktarılan sesleri dağıtan kullanıcılar sorumludur. <br>Eğer bu maddeyle aynı fikirde değilseniz, yazılım paketi içindeki herhangi bir kod veya dosyayı kullanamaz veya referans göremezsiniz. Detaylar için kök dizindeki <b>Agreement-LICENSE.txt</b> dosyasına bakınız.",
|
11 |
+
"模型推理": "Model çıkartma (Inference)",
|
12 |
+
"推理音色": "Ses çıkartma (Inference):",
|
13 |
+
"刷新音色列表和索引路径": "Ses listesini ve indeks yolunu yenile",
|
14 |
+
"卸载音色省显存": "GPU bellek kullanımını azaltmak için sesi kaldır",
|
15 |
+
"请选择说话人id": "Konuşmacı/Şarkıcı No seçin:",
|
16 |
+
"男转女推荐+12key, 女转男推荐-12key, 如果音域爆炸导致音色失真也可以自己调整到合适音域. ": "Erkekten kadına çevirmek için +12 tuş önerilir, kadından erkeğe çevirmek için ise -12 tuş önerilir. Eğer ses aralığı çok fazla genişler ve ses bozulursa, isteğe bağlı olarak uygun aralığa kendiniz de ayarlayabilirsiniz.",
|
17 |
+
"变调(整数, 半音数量, 升八度12降八度-12)": "Transpoze et (tamsayı, yarıton sayısıyla; bir oktav yükseltmek için: 12, bir oktav düşürmek için: -12):",
|
18 |
+
"输入待处理音频文件路径(默认是正确格式示例)": "İşlenecek ses dosyasının yolunu girin (varsayılan doğru format örneğidir):",
|
19 |
+
"选择音高提取算法,输入歌声可用pm提速,harvest低音好但巨慢无比,crepe效果好但吃GPU": "Pitch algoritmasını seçin ('pm': daha hızlı çıkarır ancak daha düşük kaliteli konuşma; 'harvest': daha iyi konuşma sesi ancak son derece yavaş; 'crepe': daha da iyi kalite ancak GPU yoğunluğu gerektirir):",
|
20 |
+
">=3则使用对harvest音高识别的结果使用中值滤波,数值为滤波半径,使用可以削弱哑音": "Eğer >=3 ise, elde edilen pitch sonuçlarına median filtreleme uygula. Bu değer, filtre yarıçapını temsil eder ve nefesliliği azaltabilir.",
|
21 |
+
"特征检索库文件路径,为空则使用下拉的选择结果": "Özellik indeksi dosyasının yolunu belirtin. Seçilen sonucu kullanmak için boş bırakın veya açılır menüden seçim yapın.",
|
22 |
+
"自动检测index路径,下拉式选择(dropdown)": "İndeks yolunu otomatik olarak tespit et ve açılır menüden seçim yap.",
|
23 |
+
"特征文件路径": "Özellik dosyasının yolu:",
|
24 |
+
"检索特征占比": "Arama özelliği oranı (vurgu gücünü kontrol eder, çok yüksek olması sanal etkilere neden olur)",
|
25 |
+
"后处理重采样至最终采样率,0为不进行重采样": "Son işleme aşamasında çıktı sesini son örnekleme hızına yeniden örnekle. 0 değeri için yeniden örnekleme yapılmaz:",
|
26 |
+
"输入源音量包络替换输出音量包络融合比例,越靠近1越使用输出包络": "Sesin hacim zarfını ayarlayın. 0'a yakın değerler, sesin orijinal vokallerin hacmine benzer olmasını sağlar. Düşük bir değerle ses gürültüsünü maskeleyebilir ve hacmi daha doğal bir şekilde duyulabilir hale getirebilirsiniz. 1'e yaklaştıkça sürekli bir yüksek ses seviyesi elde edilir:",
|
27 |
+
"保护清辅音和呼吸声,防止电音撕裂等artifact,拉满0.5不开启,调低加大保护力度但可能降低索引效果": "Sessiz ünsüzleri ve nefes seslerini koruyarak elektronik müzikte yırtılma gibi sanal hataların oluşmasını engeller. 0.5 olarak ayarlandığında devre dışı kalır. Değerin azaltılması korumayı artırabilir, ancak indeksleme doğruluğunu azaltabilir:",
|
28 |
+
"F0曲线文件, 可选, 一行一个音高, 代替默认F0及升降调": "F0 eğrisi dosyası (isteğe bağlı). Her satırda bir pitch değeri bulunur. Varsayılan F0 ve pitch modülasyonunu değiştirir:",
|
29 |
+
"转换": "Dönüştür",
|
30 |
+
"输出信息": "Çıkış bilgisi",
|
31 |
+
"输出音频(右下角三个点,点了可以下载)": "Ses dosyasını dışa aktar (indirmek için sağ alt köşedeki üç noktaya tıklayın)",
|
32 |
+
"批量转换, 输入待转换音频文件夹, 或上传多个音频文件, 在指定文件夹(默认opt)下输出转换的音频. ": "Toplu dönüştür. Dönüştürülecek ses dosyalarının bulunduğu klasörü girin veya birden çok ses dosyasını yükleyin. Dönüştürülen ses dosyaları belirtilen klasöre ('opt' varsayılan olarak) dönüştürülecektir",
|
33 |
+
"指定输出文件夹": "Çıkış klasörünü belirt:",
|
34 |
+
"输入待处理音频文件夹路径(去文件管理器地址栏拷就行了)": "İşlenecek ses klasörünün yolunu girin (dosya yöneticisinin adres çubuğundan kopyalayın):",
|
35 |
+
"也可批量输入音频文件, 二选一, 优先读文件夹": "Toplu olarak ses dosyalarını da girebilirsiniz. İki seçenekten birini seçin. Öncelik klasörden okumaya verilir.",
|
36 |
+
"导出文件格式": "Dışa aktarma dosya formatı",
|
37 |
+
"伴奏人声分离&去混响&去回声": "Vokal/Müzik Ayrıştırma ve Yankı Giderme",
|
38 |
+
"人声伴奏分离批量处理, 使用UVR5模型。 <br>合格的文件夹路径格式举例: E:\\codes\\py39\\vits_vc_gpu\\白鹭霜华测试样例(去文件管理器地址栏拷就行了)。 <br>模型分为三类: <br>1、保留人声:不带和声的音频选这个,对主人声保留比HP5更好。内置HP2和HP3两个模型,HP3可能轻微漏伴奏但对主人声保留比HP2稍微好一丁点; <br>2、仅保留主人声:带和声的音频选这个,对主人声可能有削弱。内置HP5一个模型; <br> 3、去混响、去延迟模型(by FoxJoy):<br> (1)MDX-Net(onnx_dereverb):对于双通道混响是最好的选择,不能去除单通道混响;<br> (234)DeEcho:去除延迟效果。Aggressive比Normal去除得更彻底,DeReverb额外去除混响,可去除单声道混响,但是对高频重的板式混响去不干净。<br>去混响/去延迟,附:<br>1、DeEcho-DeReverb模型的耗时是另外2个DeEcho模型的接近2倍;<br>2、MDX-Net-Dereverb模型挺慢的;<br>3、个人推荐的最干净的配置是先MDX-Net再DeEcho-Aggressive。": "Batch işleme kullanarak vokal eşlik ayrımı için UVR5 modeli kullanılır.<br>Geçerli bir klasör yol formatı örneği: D:\\path\\to\\input\\folder (dosya yöneticisi adres çubuğundan kopyalanır).<br>Model üç kategoriye ayrılır:<br>1. Vokalleri koru: Bu seçeneği, harmoni içermeyen sesler için kullanın. HP5'ten daha iyi bir şekilde vokalleri korur. İki dahili model içerir: HP2 ve HP3. HP3, eşlik sesini hafifçe sızdırabilir, ancak vokalleri HP2'den biraz daha iyi korur.<br>2. Sadece ana vokalleri koru: Bu seçeneği, harmoni içeren sesler için kullanın. Ana vokalleri zayıflatabilir. Bir dahili model içerir: HP5.<br>3. Reverb ve gecikme modelleri (FoxJoy tarafından):<br> (1) MDX-Net: Stereo reverb'i kaldırmak için en iyi seçenek, ancak mono reverb'i kaldıramaz;<br> (234) DeEcho: Gecikme efektlerini kaldırır. Agresif mod, Normal moda göre daha kapsamlı bir şekilde kaldırma yapar. DeReverb ayrıca reverb'i kaldırır ve mono reverb'i kaldırabilir, ancak yoğun yankılı yüksek frekanslı içerikler için çok etkili değildir.<br>Reverb/gecikme notları:<br>1. DeEcho-DeReverb modelinin işleme süresi diğer iki DeEcho modeline göre yaklaşık olarak iki kat daha uzundur.<br>2. MDX-Net-Dereverb modeli oldukça yavaştır.<br>3. Tavsiye edilen en temiz yapılandırma önce MDX-Net'i uygulamak ve ardından DeEcho-Aggressive uygulamaktır.",
|
39 |
+
"输入待处理音频文件夹路径": "İşlenecek ses klasörünün yolunu girin:",
|
40 |
+
"模型": "Model",
|
41 |
+
"指定输出主人声文件夹": "Vokal için çıkış klasörünü belirtin:",
|
42 |
+
"指定输出非主人声文件夹": "Müzik ve diğer sesler için çıkış klasörünü belirtin:",
|
43 |
+
"训练": "Eğitim",
|
44 |
+
"step1: 填写实验配置. 实验数据放在logs下, 每个实验一个文件夹, 需手工输入实验名路径, 内含实验配置, 日志, 训练得到的模型文件. ": "Adım 1: Deneysel yapılandırmayı doldurun. Deneysel veriler 'logs' klasöründe saklanır ve her bir deney için ayrı bir klasör vardır. Deneysel adı yolu manuel olarak girin; bu yol, deneysel yapılandırmayı, günlükleri ve eğitilmiş model dosyalarını içerir.",
|
45 |
+
"输入实验名": "Deneysel adı girin:",
|
46 |
+
"目标采样率": "Hedef örnekleme oranı:",
|
47 |
+
"模型是否带音高指导(唱歌一定要, 语音可以不要)": "Modelin ses yüksekliği (Pitch) rehberliği içerip içermediği (şarkı söyleme için şarttır, konuşma için isteğe bağlıdır):",
|
48 |
+
"版本": "Sürüm",
|
49 |
+
"提取音高和处理数据使用的CPU进程数": "Ses yüksekliği çıkartmak (Pitch) ve verileri işlemek için kullanılacak CPU işlemci sayısı:",
|
50 |
+
"step2a: 自动遍历训练文件夹下所有可解码成音频的文件并进行切片归一化, 在实验目录下生成2个wav文件夹; 暂时只支持单人训练. ": "Adım 2a: Eğitim klasöründe ses dosyalarını otomatik olarak gezinerek dilimleme normalizasyonu yapın. Deney dizini içinde 2 wav klasörü oluşturur. Şu anda sadece tek kişilik eğitim desteklenmektedir.",
|
51 |
+
"输入训练文件夹路径": "Eğitim klasörünün yolunu girin:",
|
52 |
+
"请指定说话人id": "Lütfen konuşmacı/sanatçı no belirtin:",
|
53 |
+
"处理数据": "Verileri işle",
|
54 |
+
"step2b: 使用CPU提取音高(如果模型带音高), 使用GPU提取特征(选择卡号)": "Adım 2b: Ses yüksekliği (Pitch) çıkartmak için CPU kullanın (eğer model ses yüksekliği içeriyorsa), özellikleri çıkartmak için GPU kullanın (GPU indeksini seçin):",
|
55 |
+
"以-分隔输入使用的卡号, 例如 0-1-2 使用卡0和卡1和卡2": "GPU indekslerini '-' ile ayırarak girin, örneğin 0-1-2, GPU 0, 1 ve 2'yi kullanmak için:",
|
56 |
+
"显卡信息": "GPU Bilgisi",
|
57 |
+
"选择音高提取算法:输入歌声可用pm提速,高质量语音但CPU差可用dio提速,harvest质量更好但慢": "Ses yüksekliği (Pitch) çıkartma algoritmasını seçin ('pm': daha hızlı çıkartma, ancak düşük kaliteli konuşma; 'dio': geliştirilmiş konuşma kalitesi, ancak daha yavaş çıkartma; 'harvest': daha iyi kalite, ancak daha da yavaş çıkartma):",
|
58 |
+
"特征提取": "Özellik çıkartma",
|
59 |
+
"step3: 填写训练设置, 开始训练模型和索引": "Adım 3: Eğitim ayarlarını doldurun ve modeli ve dizini eğitmeye başlayın",
|
60 |
+
"保存频率save_every_epoch": "Kaydetme sıklığı (save_every_epoch):",
|
61 |
+
"总训练轮数total_epoch": "Toplam eğitim turu (total_epoch):",
|
62 |
+
"每张显卡的batch_size": "Her GPU için yığın boyutu (batch_size):",
|
63 |
+
"是否仅保存最新的ckpt文件以节省硬盘空间": "Sadece en son '.ckpt' dosyasını kaydet:",
|
64 |
+
"否": "Hayır",
|
65 |
+
"是否缓存所有训练集至显存. 10min以下小数据可缓存以加速训练, 大数据缓存会炸显存也加不了多少速": "Tüm eğitim verilerini GPU belleğine önbelleğe alıp almayacağınızı belirtin. Küçük veri setlerini (10 dakikadan az) önbelleğe almak eğitimi hızlandırabilir, ancak büyük veri setlerini önbelleğe almak çok fazla GPU belleği tüketir ve çok fazla hız artışı sağlamaz:",
|
66 |
+
"是否在每次保存时间点将最终小模型保存至weights文件夹": "Her kaydetme noktasında son küçük bir modeli 'weights' klasörüne kaydetmek için:",
|
67 |
+
"加载预训练底模G路径": "Önceden eğitilmiş temel G modelini yükleme yolu:",
|
68 |
+
"加载预训练底模D路径": "Önceden eğitilmiş temel D modelini yükleme yolu:",
|
69 |
+
"训练模型": "Modeli Eğit",
|
70 |
+
"训练特征索引": "Özellik Dizinini Eğit",
|
71 |
+
"一键训练": "Tek Tuşla Eğit",
|
72 |
+
"ckpt处理": "ckpt İşleme",
|
73 |
+
"模型融合, 可用于测试音色融合": "Model birleştirme, ses rengi birleştirmesi için kullanılabilir",
|
74 |
+
"A模型路径": "A Modeli Yolu:",
|
75 |
+
"B模型路径": "B Modeli Yolu:",
|
76 |
+
"A模型权重": "A Modeli Ağırlığı:",
|
77 |
+
"模型是否带音高指导": "Modelin ses yüksekliği rehberi içerip içermediği:",
|
78 |
+
"要置入的模型信息": "Eklemek için model bilgileri:",
|
79 |
+
"保存的模型名不带后缀": "Kaydedilecek model adı (uzantı olmadan):",
|
80 |
+
"模型版本型号": "Model mimari versiyonu:",
|
81 |
+
"融合": "Birleştir",
|
82 |
+
"修改模型信息(仅支持weights文件夹下提取的小模型文件)": "Model bilgilerini düzenle (sadece 'weights' klasöründen çıkarılan küçük model dosyaları desteklenir)",
|
83 |
+
"模型路径": "Model Yolu:",
|
84 |
+
"要改的模型信息": "Düzenlenecek model bilgileri:",
|
85 |
+
"保存的文件名, 默认空为和源文件同名": "Kaydedilecek dosya adı (varsayılan: kaynak dosya ile aynı):",
|
86 |
+
"修改": "Düzenle",
|
87 |
+
"查看模型信息(仅支持weights文件夹下提取的小模型文件)": "Model bilgilerini görüntüle (sadece 'weights' klasöründen çıkarılan küçük model dosyaları desteklenir)",
|
88 |
+
"查看": "Görüntüle",
|
89 |
+
"模型提取(输入logs文件夹下大文件模型路径),适用于训一半不想训了模型没有自动提取保存小文件模型,或者想测试中间模型的情况": "Model çıkartma (büyük dosya modeli yolunu 'logs' klasöründe girin). Bu, eğitimi yarıda bırakmak istediğinizde ve manuel olarak küçük bir model dosyası çıkartmak ve kaydetmek istediğinizde veya bir ara modeli test etmek istediğinizde kullanışlıdır:",
|
90 |
+
"保存名": "Kaydetme Adı:",
|
91 |
+
"模型是否带音高指导,1是0否": "Modelin ses yüksekliği rehberi içerip içermediği (1: evet, 0: hayır):",
|
92 |
+
"提取": "Çıkart",
|
93 |
+
"Onnx导出": "Onnx Dışa Aktar",
|
94 |
+
"RVC模型路径": "RVC Model Yolu:",
|
95 |
+
"Onnx输出路径": "Onnx Dışa Aktarım Yolu:",
|
96 |
+
"导出Onnx模型": "Onnx Modeli Dışa Aktar",
|
97 |
+
"常见问题解答": "Sıkça Sorulan Sorular (SSS)",
|
98 |
+
"招募音高曲线前端编辑器": "Ses yükseklik eğrisi ön uç düzenleyicisi için işe alım",
|
99 |
+
"加开发群联系我xxxxx": "Geliştirme grubuna katılın ve benimle iletişime geçin: xxxxx",
|
100 |
+
"点击查看交流、问题反馈群号": "İletişim ve sorun geri bildirim grup numarasını görüntülemek için tıklayın",
|
101 |
+
"xxxxx": "xxxxx",
|
102 |
+
"加载模型": "Model yükle",
|
103 |
+
"Hubert模型": "Hubert Modeli",
|
104 |
+
"选择.pth文件": ".pth dosyası seç",
|
105 |
+
"选择.index文件": ".index dosyası seç",
|
106 |
+
"选择.npy文件": ".npy dosyası seç",
|
107 |
+
"输入设备": "Giriş cihazı",
|
108 |
+
"输出设备": "Çıkış cihazı",
|
109 |
+
"音频设备(请使用同种类驱动)": "Ses cihazı (aynı tür sürücüyü kullanın)",
|
110 |
+
"响应阈值": "Tepki eşiği",
|
111 |
+
"音调设置": "Pitch ayarları",
|
112 |
+
"Index Rate": "Index Oranı",
|
113 |
+
"常规设置": "Genel ayarlar",
|
114 |
+
"采样长度": "Örnekleme uzunluğu",
|
115 |
+
"淡入淡出长度": "Geçiş (Fade) uzunluğu",
|
116 |
+
"额外推理时长": "Ekstra çıkartma süresi",
|
117 |
+
"输入降噪": "Giriş gürültü azaltma",
|
118 |
+
"输出降噪": "Çıkış gürültü azaltma",
|
119 |
+
"性能设置": "Performans ayarları",
|
120 |
+
"开始音频转换": "Ses dönüştürmeyi başlat",
|
121 |
+
"停止音频转换": "Ses dönüştürmeyi durdur",
|
122 |
+
"推理时间(ms):": "Çıkarsama süresi (ms):",
|
123 |
+
"请选择pth文件": "Lütfen .pth dosyası seçin",
|
124 |
+
"请选择index文件": "Lütfen .index dosyası seçin",
|
125 |
+
"hubert模型路径不可包含中文": "hubert modeli yolu Çince karakter içeremez",
|
126 |
+
"pth文件路径不可包含中文": ".pth dosya yolu Çince karakter içeremez",
|
127 |
+
"index文件路径不可包含中文": ".index dosya yolu Çince karakter içeremez",
|
128 |
+
"音高算法": "音高算法",
|
129 |
+
"harvest进程数": "harvest进程数"
|
130 |
+
}
|
i18n/zh_CN.json
ADDED
@@ -0,0 +1,132 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"很遗憾您这没有能用的显卡来支持您训练": "很遗憾您这没有能用的显卡来支持您训练",
|
3 |
+
"是": "是",
|
4 |
+
"step1:正在处理数据": "step1:正在处理数据",
|
5 |
+
"step2a:无需提取音高": "step2a:无需提取音高",
|
6 |
+
"step2b:正在提取特征": "step2b:正在提取特征",
|
7 |
+
"step3a:正在训练模型": "step3a:正在训练模型",
|
8 |
+
"训练结束, 您可查看控制台训练日志或实验文件夹下的train.log": "训练结束, 您可查看控制台训练日志或实验文件夹下的train.log",
|
9 |
+
"全流程结束!": "全流程结束!",
|
10 |
+
"本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责. <br>如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录<b>LICENSE</b>.": "本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责. <br>如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录<b>LICENSE</b>.",
|
11 |
+
"模型推理": "模型推理",
|
12 |
+
"推理音色": "推理音色",
|
13 |
+
"刷新音色列表和索引路径": "刷新音色列表和索引路径",
|
14 |
+
"卸载音色省显存": "卸载音色省显存",
|
15 |
+
"请选择说话人id": "请选择说话人id",
|
16 |
+
"男转女推荐+12key, 女转男推荐-12key, 如果音域爆炸导致音色失真也可以自己调整到合适音域. ": "男转女推荐+12key, 女转男推荐-12key, 如果音域爆炸导致音色失真也可以自己调整到合适音域. ",
|
17 |
+
"变调(整数, 半音数量, 升八度12降八度-12)": "变调(整数, 半音数量, 升八度12降八度-12)",
|
18 |
+
"输入待处理音频文件路径(默认是正确格式示例)": "输入待处理音频文件路径(默认是正确格式示例)",
|
19 |
+
"选择音高提取算法,输入歌声可用pm提速,harvest低音好但巨慢无比,crepe效果好但吃GPU": "选择音高提取算法,输入歌声可用pm提速,harvest低音好但巨慢无比,crepe效果好但吃GPU",
|
20 |
+
"crepe_hop_length": "Crepe Hop Length (Only applies to crepe): Hop length refers to the time it takes for the speaker to jump to a dramatic pitch. Lower hop lengths take more time to infer but are more pitch accurate.",
|
21 |
+
"特征检索库文件路径": "特征检索库文件路径",
|
22 |
+
"特征检索库文件路径,为空则使用下拉的选择结果": "特征检索库文件路径,为空则使用下拉的选择结果",
|
23 |
+
">=3则使用对harvest音高识别的结果使用中值滤波,数值为滤波半径,使用可以削弱哑音": ">=3则使用对harvest音高识别的结果使用中值滤波,数值为滤波半径,使用可以削弱哑音",
|
24 |
+
"自动检测index路径,下拉式选择(dropdown)": "自动检测index路径,下拉式选择(dropdown)",
|
25 |
+
"特征文件路径": "特征文件路径",
|
26 |
+
"检索特征占比": "检索特征占比",
|
27 |
+
"后处理重采样至最终采样率,0为不进行重采样": "后处理重采样至最终采样率,0为不进行重采样",
|
28 |
+
"输入源音量包络替换输出音量包络融合比例,越靠近1越使用输出包络": "输入源音量包络替换输出音量包络融合比例,越靠近1越使用输出包络",
|
29 |
+
"保护清辅音和呼吸声,防止电音撕裂等artifact,拉满0.5不开启,调低加大保护力度但可能降低索引效果": "保护清辅音和呼吸声,防止电音撕裂等artifact,拉满0.5不开启,调低加大保护力度但可能降低索引效果",
|
30 |
+
"F0曲线文件, 可选, 一行一个音高, 代替默认F0及升降调": "F0曲线文件, 可选, 一行一个音高, 代替默认F0及升降调",
|
31 |
+
"转换": "转换",
|
32 |
+
"输出信息": "输出信息",
|
33 |
+
"输出音频(右下角三个点,点了可以下载)": "输出音频(右下角三个点,点了可以下载)",
|
34 |
+
"批量转换, 输入待转换音频文件夹, 或上传多个音频文件, 在指定文件夹(默认opt)下输出转换的音频. ": "批量转换, 输入待转换音频文件夹, 或上传多个音频文件, 在指定文件夹(默认opt)下输出转换的音频. ",
|
35 |
+
"指定输出文件夹": "指定输出文件夹",
|
36 |
+
"输入待处理音频文件夹路径(去文件管理器地址栏拷就行了)": "输入待处理音频文件夹路径(去文件管理器地址栏拷就行了)",
|
37 |
+
"也可批量输入音频文件, 二选一, 优先读文件夹": "也可批量输入音频文件, 二选一, 优先读文件夹",
|
38 |
+
"导出文件格式": "导出文件格式",
|
39 |
+
"伴奏人声分离&去混响&去回声": "伴奏人声分离&去混响&去回声",
|
40 |
+
"人声伴奏分离批量处理, 使用UVR5模型。 <br>合格的文件夹路径格式举例: E:\\codes\\py39\\vits_vc_gpu\\白鹭霜华测试样例(去文件管理器地址栏拷就行了)。 <br>模型分为三类: <br>1、保留人声:不带和声的音频选这个,对主人声保留比HP5更好。内置HP2和HP3两个模型,HP3可能轻微漏伴奏但对主人声保留比HP2稍微好一丁点; <br>2、仅保留主人声:带和声的音频选这个,对主人声可能有削弱。内置HP5一个模型; <br> 3、去混响、去延迟模型(by FoxJoy):<br> (1)MDX-Net(onnx_dereverb):对于双通道混响是最好的选择,不能去除单通道混响;<br> (234)DeEcho:去除延迟效果。Aggressive比Normal去除得更彻底,DeReverb额外去除混响,可去除单声道混响,但是对高频重的板式混响去不干净。<br>去混响/去延迟,附:<br>1、DeEcho-DeReverb模型的耗时是另外2个DeEcho模型的接近2倍;<br>2、MDX-Net-Dereverb模型挺慢的;<br>3、个人推荐的最干净的配置是先MDX-Net再DeEcho-Aggressive。": "人声伴奏分离批量处理, 使用UVR5模型。 <br>合格的文件夹路径格式举例: E:\\codes\\py39\\vits_vc_gpu\\白鹭霜华测试样例(去文件管理器地址栏拷就行了)。 <br>模型分为三类: <br>1、保留人声:不带和声的音频选这个,对主人声保留比HP5更好。内置HP2和HP3两个模型,HP3可能轻微漏伴奏但对主人声保留比HP2稍微好一丁点; <br>2、仅保留主人声:带和声的音频选这个,对主人声可能有削弱。内置HP5一个模型; <br> 3、去混响、去延迟模型(by FoxJoy):<br> (1)MDX-Net(onnx_dereverb):对于双通道混响是最好的选择,不能去除单通道混响;<br> (234)DeEcho:去除延迟效果。Aggressive比Normal去除得更彻底,DeReverb额外去除混响,可去除单声道混响,但是对高频重的板式混响去不干净。<br>去混响/去延迟,附:<br>1、DeEcho-DeReverb模型的耗时是另外2个DeEcho模型的接近2倍;<br>2、MDX-Net-Dereverb模型挺慢的;<br>3、个人推荐的最干净的配置是先MDX-Net再DeEcho-Aggressive。",
|
41 |
+
"输入待处理音频文件夹路径": "输入待处理音频文件夹路径",
|
42 |
+
"模型": "模型",
|
43 |
+
"指定输出主人声文件夹": "指定输出主人声文件夹",
|
44 |
+
"指定输出非主人声文件夹": "指定输出非主人声文件夹",
|
45 |
+
"训练": "训练",
|
46 |
+
"step1: 填写实验配置. 实验数据放在logs下, 每个实验一个文件夹, 需手工输入实验名路径, 内含实验配置, 日志, 训练得到的模型文件. ": "step1: 填写实验配置. 实验数据放在logs下, 每个实验一个文件夹, 需手工输入实验名路径, 内含实验配置, 日志, 训练得到的模型文件. ",
|
47 |
+
"输入实验名": "输入实验名",
|
48 |
+
"目标采样率": "目标采样率",
|
49 |
+
"模型是否带音高指导(唱歌一定要, 语音可以不要)": "模型是否带音高指导(唱歌一定要, 语音可以不要)",
|
50 |
+
"版本": "版本",
|
51 |
+
"提取音高和处理数据使用的CPU进程数": "提取音高和处理数据使用的CPU进程数",
|
52 |
+
"step2a: 自动遍历训练文件夹下所有可解码成音频的文件并进行切片归一化, 在实验目录下生成2个wav文件夹; 暂时只支持单人训练. ": "step2a: 自动遍历训练文件夹下所有可解码成音频的文件并进行切片归一化, 在实验目录下生成2个wav文件夹; 暂时只支持单人训练. ",
|
53 |
+
"输入训练文件夹路径": "输入训练文件夹路径",
|
54 |
+
"请指定说话人id": "请指定说话人id",
|
55 |
+
"处理数据": "处理数据",
|
56 |
+
"step2b: 使用CPU提取音高(如果模型带音高), 使用GPU提取特征(选择卡号)": "step2b: 使用CPU提取音高(如果模型带音高), 使用GPU提取特征(选择卡号)",
|
57 |
+
"以-分隔输入使用的卡号, 例如 0-1-2 使用卡0和卡1和卡2": "以-分隔输入使用的卡号, 例如 0-1-2 使用卡0和卡1和卡2",
|
58 |
+
"显卡信息": "显卡信息",
|
59 |
+
"选择音高提取算法:输入歌声可用pm提速,高质量语音但CPU差可用dio提速,harvest质量更好但慢": "选择音高提取算法:输入歌声可用pm提速,高质量语音但CPU差可用dio提速,harvest质量更好但慢",
|
60 |
+
"特征提取": "特征提取",
|
61 |
+
"step3: 填写训练设置, 开始训练模型和索引": "step3: 填写训练设置, 开始训练模型和索引",
|
62 |
+
"保存频率save_every_epoch": "保存频率save_every_epoch",
|
63 |
+
"总训练轮数total_epoch": "总训练轮数total_epoch",
|
64 |
+
"每张显卡的batch_size": "每张显卡的batch_size",
|
65 |
+
"是否仅保存最新的ckpt文件以节省硬盘空间": "是否仅保存最新的ckpt文件以节省硬盘空间",
|
66 |
+
"否": "否",
|
67 |
+
"是否缓存所有训练集至显存. 10min以下小数据可缓存以加速训练, 大数据缓存会炸显存也加不了多少速": "是否缓存所有训练集至显存. 10min以下小数据可缓存以加速训练, 大数据缓存会炸显存也加不了多少速",
|
68 |
+
"是否在每次保存时间点将最终小模型保存至weights文件夹": "是否在每次保存时间点将最终小模型保存至weights文件夹",
|
69 |
+
"加载预训练底模G路径": "加载预训练底模G路径",
|
70 |
+
"加载预训练底模D路径": "加载预训练底模D路径",
|
71 |
+
"训练模型": "训练模型",
|
72 |
+
"训练特征索引": "���练特征索引",
|
73 |
+
"一键训练": "一键训练",
|
74 |
+
"ckpt处理": "ckpt处理",
|
75 |
+
"模型融合, 可用于测试音色融合": "模型融合, 可用于测试音色融合",
|
76 |
+
"A模型路径": "A模型路径",
|
77 |
+
"B模型路径": "B模型路径",
|
78 |
+
"A模型权重": "A模型权重",
|
79 |
+
"模型是否带音高指导": "模型是否带音高指导",
|
80 |
+
"要置入的模型信息": "要置入的模型信息",
|
81 |
+
"保存的模型名不带后缀": "保存的模型名不带后缀",
|
82 |
+
"模型版本型号": "模型版本型号",
|
83 |
+
"融合": "融合",
|
84 |
+
"修改模型信息(仅支持weights文件夹下提取的小模型文件)": "修改模型信息(仅支持weights文件夹下提取的小模型文件)",
|
85 |
+
"模型路径": "模型路径",
|
86 |
+
"要改的模型信息": "要改的模型信息",
|
87 |
+
"保存的文件名, 默认空为和源文件同名": "保存的文件名, 默认空为和源文件同名",
|
88 |
+
"修改": "修改",
|
89 |
+
"查看模型信息(仅支持weights文件夹下提取的小模型文件)": "查看模型信息(仅支持weights文件夹下提取的小模型文件)",
|
90 |
+
"查看": "查看",
|
91 |
+
"模型提取(输入logs文件夹下大文件模型路径),适用于训一半不想训了模型没有自动提取保存小文件模型,或者想测试中间模型的情况": "模型提取(输入logs文件夹下大文件模型路径),适用于训一半不想训了模型没有自动提取保存小文件模型,或者想测试中间模型的情况",
|
92 |
+
"保存名": "保存名",
|
93 |
+
"模型是否带音高指导,1是0否": "模型是否带音高指导,1是0否",
|
94 |
+
"提取": "提取",
|
95 |
+
"Onnx导出": "Onnx导出",
|
96 |
+
"RVC模型路径": "RVC模型路径",
|
97 |
+
"Onnx输出路径": "Onnx输出路径",
|
98 |
+
"导出Onnx模型": "导出Onnx模型",
|
99 |
+
"常见问题解答": "常见问题解答",
|
100 |
+
"招募音高曲线前端编辑器": "招募音高曲线前端编辑器",
|
101 |
+
"加开发群联系我xxxxx": "加开发群联系我xxxxx",
|
102 |
+
"点击查看交流、问题反馈群号": "点击查看交流、问题反馈群号",
|
103 |
+
"xxxxx": "xxxxx",
|
104 |
+
"加载模型": "加载模型",
|
105 |
+
"Hubert模型": "Hubert模型",
|
106 |
+
"选择.pth文件": "选择.pth文件",
|
107 |
+
"选择.index文件": "选择.index文件",
|
108 |
+
"选择.npy文件": "选择.npy文件",
|
109 |
+
"输入设备": "输入设备",
|
110 |
+
"输出设备": "输出设备",
|
111 |
+
"音频设备(请使用同种类驱动)": "音频设备(请使用同种类驱动)",
|
112 |
+
"响应阈值": "响应阈值",
|
113 |
+
"音调设置": "音调设置",
|
114 |
+
"Index Rate": "Index Rate",
|
115 |
+
"常规设置": "常规设置",
|
116 |
+
"采样长度": "采样长度",
|
117 |
+
"淡入淡出长度": "淡入淡出长度",
|
118 |
+
"额外推理时长": "额外推理时长",
|
119 |
+
"输入降噪": "输入降噪",
|
120 |
+
"输出降噪": "输出降噪",
|
121 |
+
"性能设置": "性能设置",
|
122 |
+
"开始音频转换": "开始音频转换",
|
123 |
+
"停止音频转换": "停止音频转换",
|
124 |
+
"推理时间(ms):": "推理时间(ms):",
|
125 |
+
"请选择pth文件": "请选择pth文件",
|
126 |
+
"请选择index文件": "请选择index文件",
|
127 |
+
"hubert模型路径不可包含中文": "hubert模型路径不可包含中文",
|
128 |
+
"pth文件路径不可包含中文": "pth文件路径不可包含中文",
|
129 |
+
"index文件路径不可包含中文": "index文件路径不可包含中文",
|
130 |
+
"音高算法": "音高算法",
|
131 |
+
"harvest进程数": "harvest进程数"
|
132 |
+
}
|
i18n/zh_HK.json
ADDED
@@ -0,0 +1,132 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"很遗憾您这没有能用的显卡来支持您训练": "很遗憾您这没有能用的显卡来支持您训练",
|
3 |
+
"是": "是",
|
4 |
+
"step1:正在处理数据": "step1:正在处理数据",
|
5 |
+
"step2a:无需提取音高": "step2a:无需提取音高",
|
6 |
+
"step2b:正在提取特征": "step2b:正在提取特征",
|
7 |
+
"step3a:正在训练模型": "step3a:正在训练模型",
|
8 |
+
"训练结束, 您可查看控制台训练日志或实验文件夹下的train.log": "训练结束, 您可查看控制台训练日志或实验文件夹下的train.log",
|
9 |
+
"全流程结束!": "全流程结束!",
|
10 |
+
"本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责. <br>如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录<b>LICENSE</b>.": "本軟體以MIT協議開源,作者不對軟體具備任何控制力,使用軟體者、傳播軟體導出的聲音者自負全責。<br>如不認可該條款,則不能使用或引用軟體包內任何程式碼和檔案。詳見根目錄<b>使用需遵守的協議-LICENSE.txt</b>。",
|
11 |
+
"模型推理": "模型推理",
|
12 |
+
"推理音色": "推理音色",
|
13 |
+
"刷新音色列表和索引路径": "刷新音色列表和索引路徑",
|
14 |
+
"卸载音色省显存": "卸載音色節省 VRAM",
|
15 |
+
"请选择说话人id": "請選擇說話人ID",
|
16 |
+
"男转女推荐+12key, 女转男推荐-12key, 如果音域爆炸导致音色失真也可以自己调整到合适音域. ": "男性轉女性推薦+12key,女性轉男性推薦-12key,如果音域爆炸導致音色失真也可以自己調整到合適音域。",
|
17 |
+
"变调(整数, 半音数量, 升八度12降八度-12)": "變調(整數、半音數量、升八度12降八度-12)",
|
18 |
+
"输入待处理音频文件路径(默认是正确格式示例)": "輸入待處理音頻檔案路徑(預設是正確格式示例)",
|
19 |
+
"选择音高提取算法,输入歌声可用pm提速,harvest低音好但巨慢无比,crepe效果好但吃GPU": "選擇音高提取演算法,輸入歌聲可用pm提速,harvest低音好但巨慢無比,crepe效果好但吃GPU",
|
20 |
+
"crepe_hop_length": "Crepe Hop Length (Only applies to crepe): Hop length refers to the time it takes for the speaker to jump to a dramatic pitch. Lower hop lengths take more time to infer but are more pitch accurate.",
|
21 |
+
"特征检索库文件路径": "特徵檢索庫檔案路徑",
|
22 |
+
"自动检测index路径,下拉式选择(dropdown)": "自動檢測index路徑,下拉式選擇(dropdown)",
|
23 |
+
">=3则使用对harvest音高识别的结果使用中值滤波,数值为滤波半径,使用可以削弱哑音": ">=3則使用對harvest音高識別的結果使用中值濾波,數值為濾波半徑,使用可以削弱啞音",
|
24 |
+
"特征检索库文件路径,为空则使用下拉的选择结果": "特徵檢索庫檔路徑,為空則使用下拉的選擇結果",
|
25 |
+
"特征文件路径": "特徵檔案路徑",
|
26 |
+
"检索特征占比": "檢索特徵佔比",
|
27 |
+
"后处理重采样至最终采样率,0为不进行重采样": "後處理重採樣至最終採樣率,0為不進行重採樣",
|
28 |
+
"输入源音量包络替换输出音量包络融合比例,越靠近1越使用输出包络": "輸入源音量包絡替換輸出音量包絡融合比例,越靠近1越使用輸出包絡",
|
29 |
+
"保护清辅音和呼吸声,防止电音撕裂等artifact,拉满0.5不开启,调低加大保护力度但可能降低索引效果": "保護清輔音和呼吸聲,防止電音撕裂等artifact,拉滿0.5不開啟,調低加大保護力度但可能降低索引效果",
|
30 |
+
"F0曲线文件, 可选, 一行一个音高, 代替默认F0及升降调": "F0曲線檔案,可選,一行一個音高,代替預設的F0及升降調",
|
31 |
+
"转换": "轉換",
|
32 |
+
"输出信息": "輸出訊息",
|
33 |
+
"输出音频(右下角三个点,点了可以下载)": "輸出音頻(右下角三個點,點了可以下載)",
|
34 |
+
"批量转换, 输入待转换音频文件夹, 或上传多个音频文件, 在指定文件夹(默认opt)下输出转换的音频. ": "批量轉換,輸入待轉換音頻資料夾,或上傳多個音頻檔案,在指定資料夾(默認opt)下輸出轉換的音頻。",
|
35 |
+
"指定输出文件夹": "指定輸出資料夾",
|
36 |
+
"输入待处理音频文件夹路径(去文件管理器地址栏拷就行了)": "輸入待處理音頻資料夾路徑(去檔案管理器地址欄拷貝即可)",
|
37 |
+
"也可批量输入音频文件, 二选一, 优先读文件夹": "也可批量輸入音頻檔案,二選一,優先讀資料夾",
|
38 |
+
"导出文件格式": "導出檔格式",
|
39 |
+
"伴奏人声分离&去混响&去回声": "伴奏人聲分離&去混響&去回聲",
|
40 |
+
"人声伴奏分离批量处理, 使用UVR5模型。 <br>合格的文件夹路径格式举例: E:\\codes\\py39\\vits_vc_gpu\\白鹭霜华测试样例(去文件管理器地址栏拷就行了)。 <br>模型分为三类: <br>1、保留人声:不带和声的音频选这个,对主人声保留比HP5更好。��置HP2和HP3两个模型,HP3可能轻微漏伴奏但对主人声保留比HP2稍微好一丁点; <br>2、仅保留主人声:带和声的音频选这个,对主人声可能有削弱。内置HP5一个模型; <br> 3、去混响、去延迟模型(by FoxJoy):<br> (1)MDX-Net(onnx_dereverb):对于双通道混响是最好的选择,不能去除单通道混响;<br> (234)DeEcho:去除延迟效果。Aggressive比Normal去除得更彻底,DeReverb额外去除混响,可去除单声道混响,但是对高频重的板式混响去不干净。<br>去混响/去延迟,附:<br>1、DeEcho-DeReverb模型的耗时是另外2个DeEcho模型的接近2倍;<br>2、MDX-Net-Dereverb模型挺慢的;<br>3、个人推荐的最干净的配置是先MDX-Net再DeEcho-Aggressive。": "使用UVR5模型進行人聲伴奏分離的批次處理。<br>有效資料夾路徑格式的例子:D:\\path\\to\\input\\folder(從檔案管理員地址欄複製)。<br>模型分為三類:<br>1. 保留人聲:選擇這個選項適用於沒有和聲的音訊。它比HP5更好地保留了人聲。它包括兩個內建模型:HP2和HP3。HP3可能輕微漏出伴奏,但比HP2更好地保留了人聲;<br>2. 僅保留主人聲:選擇這個選項適用於有和聲的音訊。它可能會削弱主人聲。它包括一個內建模型:HP5。<br>3. 消除混響和延遲模型(由FoxJoy提供):<br> (1) MDX-Net:對於立體聲混響的移除是最好的選擇,但不能移除單聲道混響;<br> (234) DeEcho:移除延遲效果。Aggressive模式比Normal模式移除得更徹底。DeReverb另外移除混響,可以移除單聲道混響,但對於高頻重的板式混響移除不乾淨。<br>消除混響/延遲注意事項:<br>1. DeEcho-DeReverb模型的處理時間是其他兩個DeEcho模型的近兩倍;<br>2. MDX-Net-Dereverb模型相當慢;<br>3. 個人推薦的最乾淨配置是先使用MDX-Net,然後使用DeEcho-Aggressive。",
|
41 |
+
"输入待处理音频文件夹路径": "輸入待處理音頻資料夾路徑",
|
42 |
+
"模型": "模型",
|
43 |
+
"指定输出主人声文件夹": "指定输出主人声文件夹",
|
44 |
+
"指定输出非主人声文件夹": "指定输出非主人声文件夹",
|
45 |
+
"训练": "訓練",
|
46 |
+
"step1: 填写实验配置. 实验数据放在logs下, 每个实验一个文件夹, 需手工输入实验名路径, 内含实验配置, 日志, 训练得到的模型文件. ": "step1:填寫實驗配置。實驗數據放在logs下,每個實驗一個資料夾,需手動輸入實驗名路徑,內含實驗配置、日誌、訓練得到的模型檔案。",
|
47 |
+
"输入实验名": "輸入實驗名稱",
|
48 |
+
"目标采样率": "目標取樣率",
|
49 |
+
"模型是否带音高指导(唱歌一定要, 语音可以不要)": "模型是否帶音高指導(唱歌一定要,語音可以不要)",
|
50 |
+
"版本": "版本",
|
51 |
+
"提取音高和处理数据使用的CPU进程数": "提取音高和處理數據使用的CPU進程數",
|
52 |
+
"step2a: 自动遍历训练文件夹下所有可解码成音频的文件并进行切片归一化, 在实验目录下生成2个wav文件夹; 暂时只支持单人训练. ": "step2a:自動遍歷訓練資料夾下所有可解碼成音頻的檔案並進行切片歸一化,在實驗目錄下生成2個wav資料夾;暫時只支援單人訓練。",
|
53 |
+
"输入训练文件夹路径": "輸入訓練檔案夾路徑",
|
54 |
+
"请指定说话人id": "請指定說話人id",
|
55 |
+
"处理数据": "處理資料",
|
56 |
+
"step2b: 使用CPU提取音高(如果模型带音高), 使用GPU提取特征(选择卡号)": "步驟2b: 使用CPU提取音高(如果模型帶音高), 使用GPU提取特徵(選擇卡號)",
|
57 |
+
"以-分隔输入使用的卡号, 例如 0-1-2 使用卡0和卡1和卡2": "以-分隔輸入使用的卡號, 例如 0-1-2 使用卡0和卡1和卡2",
|
58 |
+
"显卡信息": "顯示卡資訊",
|
59 |
+
"选择音高提取算法:输入歌声可用pm提速,高质量语音但CPU差可用dio提速,harvest质量更好但慢": "選擇音高提取算法:輸入歌聲可用pm提速,高品質語音但CPU差可用dio提速,harvest品質更好但較慢",
|
60 |
+
"特征提取": "特徵提取",
|
61 |
+
"step3: 填写训练设置, 开始训练模型和索引": "步驟3: 填寫訓練設定, 開始訓練模型和索引",
|
62 |
+
"保存频率save_every_epoch": "保存頻率save_every_epoch",
|
63 |
+
"总训练轮数total_epoch": "總訓練輪數total_epoch",
|
64 |
+
"每张显卡的batch_size": "每张显卡的batch_size",
|
65 |
+
"是否仅保存最新的ckpt文件以节省硬盘空间": "是否僅保存最新的ckpt檔案以節省硬碟空間",
|
66 |
+
"否": "否",
|
67 |
+
"是否缓存所有训练集至显存. 10min以下小数据可缓存以加速训练, 大数据缓存会炸显存也加不了多少速": "是否緩存所有訓練集至 VRAM。小於10分鐘的小數據可緩存以加速訓練,大數據緩存會爆 VRAM 也加不了多少速度",
|
68 |
+
"是否在每次保存时间点将最终小模型保存至weights文件夹": "是否在每次保存時間點將最終小模型保存至weights檔夾",
|
69 |
+
"加载预训练��模G路径": "加載預訓練底模G路徑",
|
70 |
+
"加载预训练底模D路径": "加載預訓練底模D路徑",
|
71 |
+
"训练模型": "訓練模型",
|
72 |
+
"训练特征索引": "訓練特徵索引",
|
73 |
+
"一键训练": "一鍵訓練",
|
74 |
+
"ckpt处理": "ckpt處理",
|
75 |
+
"模型融合, 可用于测试音色融合": "模型融合,可用於測試音色融合",
|
76 |
+
"A模型路径": "A模型路徑",
|
77 |
+
"B模型路径": "B模型路徑",
|
78 |
+
"A模型权重": "A模型權重",
|
79 |
+
"模型是否带音高指导": "模型是否帶音高指導",
|
80 |
+
"要置入的模型信息": "要置入的模型資訊",
|
81 |
+
"保存的模型名不带后缀": "儲存的模型名不帶副檔名",
|
82 |
+
"模型版本型号": "模型版本型號",
|
83 |
+
"融合": "融合",
|
84 |
+
"修改模型信息(仅支持weights文件夹下提取的小模型文件)": "修改模型資訊(僅支援weights資料夾下提取的小模型檔案)",
|
85 |
+
"模型路径": "模型路徑",
|
86 |
+
"要改的模型信息": "要改的模型資訊",
|
87 |
+
"保存的文件名, 默认空为和源文件同名": "儲存的檔案名,預設空為與來源檔案同名",
|
88 |
+
"修改": "修改",
|
89 |
+
"查看模型信息(仅支持weights文件夹下提取的小模型文件)": "查看模型資訊(僅支援weights資料夾下提取的小模型檔案)",
|
90 |
+
"查看": "查看",
|
91 |
+
"模型提取(输入logs文件夹下大文件模型路径),适用于训一半不想训了模型没有自动提取保存小文件模型,或者想测试中间模型的情况": "模型提取(輸入logs資料夾下大檔案模型路徑),適用於訓一半不想訓了模型沒有自動提取儲存小檔案模型,或者想測試中間模型的情況",
|
92 |
+
"保存名": "儲存名",
|
93 |
+
"模型是否带音高指导,1是0否": "模型是否帶音高指導,1是0否",
|
94 |
+
"提取": "提取",
|
95 |
+
"Onnx导出": "Onnx导出",
|
96 |
+
"RVC模型路径": "RVC模型路径",
|
97 |
+
"Onnx输出路径": "Onnx输出路径",
|
98 |
+
"导出Onnx模型": "导出Onnx模型",
|
99 |
+
"常见问题解答": "常見問題解答",
|
100 |
+
"招募音高曲线前端编辑器": "招募音高曲線前端編輯器",
|
101 |
+
"加开发群联系我xxxxx": "加開發群聯繫我xxxxx",
|
102 |
+
"点击查看交流、问题反馈群号": "點擊查看交流、問題反饋群號",
|
103 |
+
"xxxxx": "xxxxx",
|
104 |
+
"加载模型": "載入模型",
|
105 |
+
"Hubert模型": "Hubert 模型",
|
106 |
+
"选择.pth文件": "選擇 .pth 檔案",
|
107 |
+
"选择.index文件": "選擇 .index 檔案",
|
108 |
+
"选择.npy文件": "選擇 .npy 檔案",
|
109 |
+
"输入设备": "輸入設備",
|
110 |
+
"输出设备": "輸出設備",
|
111 |
+
"音频设备(请使用同种类驱动)": "音訊設備 (請使用同種類驅動)",
|
112 |
+
"响应阈值": "響應閾值",
|
113 |
+
"音调设置": "音調設定",
|
114 |
+
"Index Rate": "Index Rate",
|
115 |
+
"常规设置": "一般設定",
|
116 |
+
"采样长度": "取樣長度",
|
117 |
+
"淡入淡出长度": "淡入淡出長度",
|
118 |
+
"额外推理时长": "額外推理時長",
|
119 |
+
"输入降噪": "輸入降噪",
|
120 |
+
"输出降噪": "輸出降噪",
|
121 |
+
"性能设置": "效能設定",
|
122 |
+
"开始音频转换": "開始音訊轉換",
|
123 |
+
"停止音频转换": "停止音訊轉換",
|
124 |
+
"推理时间(ms):": "推理時間(ms):",
|
125 |
+
"请选择pth文件": "请选择pth文件",
|
126 |
+
"请选择index文件": "请选择index文件",
|
127 |
+
"hubert模型路径不可包含中文": "hubert模型路径不可包含中文",
|
128 |
+
"pth文件路径不可包含中文": "pth文件路径不可包含中文",
|
129 |
+
"index文件路径不可包含中文": "index文件路径不可包含中文",
|
130 |
+
"音高算法": "音高算法",
|
131 |
+
"harvest进程数": "harvest进程数"
|
132 |
+
}
|
i18n/zh_SG.json
ADDED
@@ -0,0 +1,132 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"很遗憾您这没有能用的显卡来支持您训练": "很遗憾您这没有能用的显卡来支持您训练",
|
3 |
+
"是": "是",
|
4 |
+
"step1:正在处理数据": "step1:正在处理数据",
|
5 |
+
"step2a:无需提取音高": "step2a:无需提取音高",
|
6 |
+
"step2b:正在提取特征": "step2b:正在提取特征",
|
7 |
+
"step3a:正在训练模型": "step3a:正在训练模型",
|
8 |
+
"训练结束, 您可查看控制台训练日志或实验文件夹下的train.log": "训练结束, 您可查看控制台训练日志或实验文件夹下的train.log",
|
9 |
+
"全流程结束!": "全流程结束!",
|
10 |
+
"本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责. <br>如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录<b>LICENSE</b>.": "本軟體以MIT協議開源,作者不對軟體具備任何控制力,使用軟體者、傳播軟體導出的聲音者自負全責。<br>如不認可該條款,則不能使用或引用軟體包內任何程式碼和檔案。詳見根目錄<b>使用需遵守的協議-LICENSE.txt</b>。",
|
11 |
+
"模型推理": "模型推理",
|
12 |
+
"推理音色": "推理音色",
|
13 |
+
"刷新音色列表和索引路径": "刷新音色列表和索引路徑",
|
14 |
+
"卸载音色省显存": "卸載音色節省 VRAM",
|
15 |
+
"请选择说话人id": "請選擇說話人ID",
|
16 |
+
"男转女推荐+12key, 女转男推荐-12key, 如果音域爆炸导致音色失真也可以自己调整到合适音域. ": "男性轉女性推薦+12key,女性轉男性推薦-12key,如果音域爆炸導致音色失真也可以自己調整到合適音域。",
|
17 |
+
"变调(整数, 半音数量, 升八度12降八度-12)": "變調(整數、半音數量、升八度12降八度-12)",
|
18 |
+
"输入待处理音频文件路径(默认是正确格式示例)": "輸入待處理音頻檔案路徑(預設是正確格式示例)",
|
19 |
+
"选择音高提取算法,输入歌声可用pm提速,harvest低音好但巨慢无比,crepe效果好但吃GPU": "選擇音高提取演算法,輸入歌聲可用pm提速,harvest低音好但巨慢無比,crepe效果好但吃GPU",
|
20 |
+
"crepe_hop_length": "Crepe Hop Length (Only applies to crepe): Hop length refers to the time it takes for the speaker to jump to a dramatic pitch. Lower hop lengths take more time to infer but are more pitch accurate.",
|
21 |
+
"特征检索库文件路径": "特徵檢索庫檔案路徑",
|
22 |
+
"特征文件路径": "特徵檔案路徑",
|
23 |
+
">=3则使用对harvest音高识别的结果使用中值滤波,数值为滤波半径,使用可以削弱哑音": ">=3則使用對harvest音高識別的結果使用中值濾波,數值為濾波半徑,使用可以削弱啞音",
|
24 |
+
"特征检索库文件路径,为空则使用下拉的选择结果": "特徵檢索庫檔路徑,為空則使用下拉的選擇結果",
|
25 |
+
"自动检测index路径,下拉式选择(dropdown)": "自動檢測index路徑,下拉式選擇(dropdown)",
|
26 |
+
"检索特征占比": "檢索特徵佔比",
|
27 |
+
"后处理重采样至最终采样率,0为不进行重采样": "後處理重採樣至最終採樣率,0為不進行重採樣",
|
28 |
+
"输入源音量包络替换输出音量包络融合比例,越靠近1越使用输出包络": "輸入源音量包絡替換輸出音量包絡融合比例,越靠近1越使用輸出包絡",
|
29 |
+
"保护清辅音和呼吸声,防止电音撕裂等artifact,拉满0.5不开启,调低加大保护力度但可能降低索引效果": "保護清輔音和呼吸聲,防止電音撕裂等artifact,拉滿0.5不開啟,調低加大保護力度但可能降低索引效果",
|
30 |
+
"F0曲线文件, 可选, 一行一个音高, 代替默认F0及升降调": "F0曲線檔案,可選,一行一個音高,代替預設的F0及升降調",
|
31 |
+
"转换": "轉換",
|
32 |
+
"输出信息": "輸出訊息",
|
33 |
+
"输出音频(右下角三个点,点了可以下载)": "輸出音頻(右下角三個點,點了可以下載)",
|
34 |
+
"批量转换, 输入待转换音频文件夹, 或上传多个音频文件, 在指定文件夹(默认opt)下输出转换的音频. ": "批量轉換,輸入待轉換音頻資料夾,或上傳多個音頻檔案,在指定資料夾(默認opt)下輸出轉換的音頻。",
|
35 |
+
"指定输出文件夹": "指定輸出資料夾",
|
36 |
+
"输入待处理音频文件夹路径(去文件管理器地址栏拷就行了)": "輸入待處理音頻資料夾路徑(去檔案管理器地址欄拷貝即可)",
|
37 |
+
"也可批量输入音频文件, 二选一, 优先读文件夹": "也可批量輸入音頻檔案,二選一,優先讀資料夾",
|
38 |
+
"导出文件格式": "導出檔格式",
|
39 |
+
"伴奏人声分离&去混响&去回声": "伴奏人聲分離&去混響&去回聲",
|
40 |
+
"人声伴奏分离批量处理, 使用UVR5模型。 <br>合格的文件夹路径格式举例: E:\\codes\\py39\\vits_vc_gpu\\白鹭霜华测试样例(去文件管理器地址栏拷就行了)。 <br>模型分为三类: <br>1、保留人声:不带和声的音频选这个,对主人声保留比HP5更好。��置HP2和HP3两个模型,HP3可能轻微漏伴奏但对主人声保留比HP2稍微好一丁点; <br>2、仅保留主人声:带和声的音频选这个,对主人声可能有削弱。内置HP5一个模型; <br> 3、去混响、去延迟模型(by FoxJoy):<br> (1)MDX-Net(onnx_dereverb):对于双通道混响是最好的选择,不能去除单通道混响;<br> (234)DeEcho:去除延迟效果。Aggressive比Normal去除得更彻底,DeReverb额外去除混响,可去除单声道混响,但是对高频重的板式混响去不干净。<br>去混响/去延迟,附:<br>1、DeEcho-DeReverb模型的耗时是另外2个DeEcho模型的接近2倍;<br>2、MDX-Net-Dereverb模型挺慢的;<br>3、个人推荐的最干净的配置是先MDX-Net再DeEcho-Aggressive。": "使用UVR5模型進行人聲伴奏分離的批次處理。<br>有效資料夾路徑格式的例子:D:\\path\\to\\input\\folder(從檔案管理員地址欄複製)。<br>模型分為三類:<br>1. 保留人聲:選擇這個選項適用於沒有和聲的音訊。它比HP5更好地保留了人聲。它包括兩個內建模型:HP2和HP3。HP3可能輕微漏出伴奏,但比HP2更好地保留了人聲;<br>2. 僅保留主人聲:選擇這個選項適用於有和聲的音訊。它可能會削弱主人聲。它包括一個內建模型:HP5。<br>3. 消除混響和延遲模型(由FoxJoy提供):<br> (1) MDX-Net:對於立體聲混響的移除是最好的選擇,但不能移除單聲道混響;<br> (234) DeEcho:移除延遲效果。Aggressive模式比Normal模式移除得更徹底。DeReverb另外移除混響,可以移除單聲道混響,但對於高頻重的板式混響移除不乾淨。<br>消除混響/延遲注意事項:<br>1. DeEcho-DeReverb模型的處理時間是其他兩個DeEcho模型的近兩倍;<br>2. MDX-Net-Dereverb模型相當慢;<br>3. 個人推薦的最乾淨配置是先使用MDX-Net,然後使用DeEcho-Aggressive。",
|
41 |
+
"输入待处理音频文件夹路径": "輸入待處理音頻資料夾路徑",
|
42 |
+
"模型": "模型",
|
43 |
+
"指定输出主人声文件夹": "指定输出主人声文件夹",
|
44 |
+
"指定输出非主人声文件夹": "指定输出非主人声文件夹",
|
45 |
+
"训练": "訓練",
|
46 |
+
"step1: 填写实验配置. 实验数据放在logs下, 每个实验一个文件夹, 需手工输入实验名路径, 内含实验配置, 日志, 训练得到的模型文件. ": "step1:填寫實驗配置。實驗數據放在logs下,每個實驗一個資料夾,需手動輸入實驗名路徑,內含實驗配置、日誌、訓練得到的模型檔案。",
|
47 |
+
"输入实验名": "輸入實驗名稱",
|
48 |
+
"目标采样率": "目標取樣率",
|
49 |
+
"模型是否带音高指导(唱歌一定要, 语音可以不要)": "模型是否帶音高指導(唱歌一定要,語音可以不要)",
|
50 |
+
"版本": "版本",
|
51 |
+
"提取音高和处理数据使用的CPU进程数": "提取音高和處理數據使用的CPU進程數",
|
52 |
+
"step2a: 自动遍历训练文件夹下所有可解码成音频的文件并进行切片归一化, 在实验目录下生成2个wav文件夹; 暂时只支持单人训练. ": "step2a:自動遍歷訓練資料夾下所有可解碼成音頻的檔案並進行切片歸一化,在實驗目錄下生成2個wav資料夾;暫時只支援單人訓練。",
|
53 |
+
"输入训练文件夹路径": "輸入訓練檔案夾路徑",
|
54 |
+
"请指定说话人id": "請指定說話人id",
|
55 |
+
"处理数据": "處理資料",
|
56 |
+
"step2b: 使用CPU提取音高(如果模型带音高), 使用GPU提取特征(选择卡号)": "步驟2b: 使用CPU提取音高(如果模型帶音高), 使用GPU提取特徵(選擇卡號)",
|
57 |
+
"以-分隔输入使用的卡号, 例如 0-1-2 使用卡0和卡1和卡2": "以-分隔輸入使用的卡號, 例如 0-1-2 使用卡0和卡1和卡2",
|
58 |
+
"显卡信息": "顯示卡資訊",
|
59 |
+
"选择音高提取算法:输入歌声可用pm提速,高质量语音但CPU差可用dio提速,harvest质量更好但慢": "選擇音高提取算法:輸入歌聲可用pm提速,高品質語音但CPU差可用dio提速,harvest品質更好但較慢",
|
60 |
+
"特征提取": "特徵提取",
|
61 |
+
"step3: 填写训练设置, 开始训练模型和索引": "步驟3: 填寫訓練設定, 開始訓練模型和索引",
|
62 |
+
"保存频率save_every_epoch": "保存頻率save_every_epoch",
|
63 |
+
"总训练轮数total_epoch": "總訓練輪數total_epoch",
|
64 |
+
"每张显卡的batch_size": "每张显卡的batch_size",
|
65 |
+
"是否仅保存最新的ckpt文件以节省硬盘空间": "是否僅保存最新的ckpt檔案以節省硬碟空間",
|
66 |
+
"否": "否",
|
67 |
+
"是否缓存所有训练集至显存. 10min以下小数据可缓存以加速训练, 大数据缓存会炸显存也加不了多少速": "是否緩存所有訓練集至 VRAM。小於10分鐘的小數據可緩存以加速訓練,大數據緩存會爆 VRAM 也加不了多少速度",
|
68 |
+
"是否在每次保存时间点将最终小模型保存至weights文件夹": "是否在每次保存時間點將最終小模型保存至weights檔夾",
|
69 |
+
"加载预训练��模G路径": "加載預訓練底模G路徑",
|
70 |
+
"加载预训练底模D路径": "加載預訓練底模D路徑",
|
71 |
+
"训练模型": "訓練模型",
|
72 |
+
"训练特征索引": "訓練特徵索引",
|
73 |
+
"一键训练": "一鍵訓練",
|
74 |
+
"ckpt处理": "ckpt處理",
|
75 |
+
"模型融合, 可用于测试音色融合": "模型融合,可用於測試音色融合",
|
76 |
+
"A模型路径": "A模型路徑",
|
77 |
+
"B模型路径": "B模型路徑",
|
78 |
+
"A模型权重": "A模型權重",
|
79 |
+
"模型是否带音高指导": "模型是否帶音高指導",
|
80 |
+
"要置入的模型信息": "要置入的模型資訊",
|
81 |
+
"保存的模型名不带后缀": "儲存的模型名不帶副檔名",
|
82 |
+
"模型版本型号": "模型版本型號",
|
83 |
+
"融合": "融合",
|
84 |
+
"修改模型信息(仅支持weights文件夹下提取的小模型文件)": "修改模型資訊(僅支援weights資料夾下提取的小模型檔案)",
|
85 |
+
"模型路径": "模型路徑",
|
86 |
+
"要改的模型信息": "要改的模型資訊",
|
87 |
+
"保存的文件名, 默认空为和源文件同名": "儲存的檔案名,預設空為與來源檔案同名",
|
88 |
+
"修改": "修改",
|
89 |
+
"查看模型信息(仅支持weights文件夹下提取的小模型文件)": "查看模型資訊(僅支援weights資料夾下提取的小模型檔案)",
|
90 |
+
"查看": "查看",
|
91 |
+
"模型提取(输入logs文件夹下大文件模型路径),适用于训一半不想训了模型没有自动提取保存小文件模型,或者想测试中间模型的情况": "模型提取(輸入logs資料夾下大檔案模型路徑),適用於訓一半不想訓了模型沒有自動提取儲存小檔案模型,或者想測試中間模型的情況",
|
92 |
+
"保存名": "儲存名",
|
93 |
+
"模型是否带音高指导,1是0否": "模型是否帶音高指導,1是0否",
|
94 |
+
"提取": "提取",
|
95 |
+
"Onnx导出": "Onnx导出",
|
96 |
+
"RVC模型路径": "RVC模型路径",
|
97 |
+
"Onnx输出路径": "Onnx输出路径",
|
98 |
+
"导出Onnx模型": "导出Onnx模型",
|
99 |
+
"常见问题解答": "常見問題解答",
|
100 |
+
"招募音高曲线前端编辑器": "招募音高曲線前端編輯器",
|
101 |
+
"加开发群联系我xxxxx": "加開發群聯繫我xxxxx",
|
102 |
+
"点击查看交流、问题反馈群号": "點擊查看交流、問題反饋群號",
|
103 |
+
"xxxxx": "xxxxx",
|
104 |
+
"加载模型": "載入模型",
|
105 |
+
"Hubert模型": "Hubert 模型",
|
106 |
+
"选择.pth文件": "選擇 .pth 檔案",
|
107 |
+
"选择.index文件": "選擇 .index 檔案",
|
108 |
+
"选择.npy文件": "選擇 .npy 檔案",
|
109 |
+
"输入设备": "輸入設備",
|
110 |
+
"输出设备": "輸出設備",
|
111 |
+
"音频设备(请使用同种类驱动)": "音訊設備 (請使用同種類驅動)",
|
112 |
+
"响应阈值": "響應閾值",
|
113 |
+
"音调设置": "音調設定",
|
114 |
+
"Index Rate": "Index Rate",
|
115 |
+
"常规设置": "一般設定",
|
116 |
+
"采样长度": "取樣長度",
|
117 |
+
"淡入淡出长度": "淡入淡出長度",
|
118 |
+
"额外推理时长": "額外推理時長",
|
119 |
+
"输入降噪": "輸入降噪",
|
120 |
+
"输出降噪": "輸出降噪",
|
121 |
+
"性能设置": "效能設定",
|
122 |
+
"开始音频转换": "開始音訊轉換",
|
123 |
+
"停止音频转换": "停止音訊轉換",
|
124 |
+
"推理时间(ms):": "推理時間(ms):",
|
125 |
+
"请选择pth文件": "请选择pth文件",
|
126 |
+
"请选择index文件": "请选择index文件",
|
127 |
+
"hubert模型路径不可包含中文": "hubert模型路径不可包含中文",
|
128 |
+
"pth文件路径不可包含中文": "pth文件路径不可包含中文",
|
129 |
+
"index文件路径不可包含中文": "index文件路径不可包含中文",
|
130 |
+
"音高算法": "音高算法",
|
131 |
+
"harvest进程数": "harvest进程数"
|
132 |
+
}
|
i18n/zh_TW.json
ADDED
@@ -0,0 +1,132 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"很遗憾您这没有能用的显卡来支持您训练": "很遗憾您这没有能用的显卡来支持您训练",
|
3 |
+
"是": "是",
|
4 |
+
"step1:正在处理数据": "step1:正在处理数据",
|
5 |
+
"step2a:无需提取音高": "step2a:无需提取音高",
|
6 |
+
"step2b:正在提取特征": "step2b:正在提取特征",
|
7 |
+
"step3a:正在训练模型": "step3a:正在训练模型",
|
8 |
+
"训练结束, 您可查看控制台训练日志或实验文件夹下的train.log": "训练结束, 您可查看控制台训练日志或实验文件夹下的train.log",
|
9 |
+
"全流程结束!": "全流程结束!",
|
10 |
+
"本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责. <br>如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录<b>LICENSE</b>.": "本軟體以MIT協議開源,作者不對軟體具備任何控制力,使用軟體者、傳播軟體導出的聲音者自負全責。<br>如不認可該條款,則不能使用或引用軟體包內任何程式碼和檔案。詳見根目錄<b>使用需遵守的協議-LICENSE.txt</b>。",
|
11 |
+
"模型推理": "模型推理",
|
12 |
+
"推理音色": "推理音色",
|
13 |
+
"刷新音色列表和索引路径": "刷新音色列表和索引路徑",
|
14 |
+
"卸载音色省显存": "卸載音色節省 VRAM",
|
15 |
+
"请选择说话人id": "請選擇說話人ID",
|
16 |
+
"男转女推荐+12key, 女转男推荐-12key, 如果音域爆炸导致音色失真也可以自己调整到合适音域. ": "男性轉女性推薦+12key,女性轉男性推薦-12key,如果音域爆炸導致音色失真也可以自己調整到合適音域。",
|
17 |
+
"变调(整数, 半音数量, 升八度12降八度-12)": "變調(整數、半音數量、升八度12降八度-12)",
|
18 |
+
"输入待处理音频文件路径(默认是正确格式示例)": "輸入待處理音頻檔案路徑(預設是正確格式示例)",
|
19 |
+
"选择音高提取算法,输入歌声可用pm提速,harvest低音好但巨慢无比,crepe效果好但吃GPU": "選擇音高提取演算法,輸入歌聲可用pm提速,harvest低音好但巨慢無比,crepe效果好但吃GPU",
|
20 |
+
"crepe_hop_length": "Crepe Hop Length (Only applies to crepe): Hop length refers to the time it takes for the speaker to jump to a dramatic pitch. Lower hop lengths take more time to infer but are more pitch accurate.",
|
21 |
+
"特征检索库文件路径": "特徵檢索庫檔案路徑",
|
22 |
+
"特征文件路径": "特徵檔案路徑",
|
23 |
+
"自动检测index路径,下拉式选择(dropdown)": "自動檢測index路徑,下拉式選擇(dropdown)",
|
24 |
+
">=3则使用对harvest音高识别的结果使用中值滤波,数值为滤波半径,使用可以削弱哑音": ">=3則使用對harvest音高識別的結果使用中值濾波,數值為濾波半徑,使用可以削弱啞音",
|
25 |
+
"特征检索库文件路径,为空则使用下拉的选择结果": "特徵檢索庫檔路徑,為空則使用下拉的選擇結果",
|
26 |
+
"检索特征占比": "檢索特徵佔比",
|
27 |
+
"后处理重采样至最终采样率,0为不进行重采样": "後處理重採樣至最終採樣率,0為不進行重採樣",
|
28 |
+
"输入源音量包络替换输出音量包络融合比例,越靠近1越使用输出包络": "輸入源音量包絡替換輸出音量包絡融合比例,越靠近1越使用輸出包絡",
|
29 |
+
"保护清辅音和呼吸声,防止电音撕裂等artifact,拉满0.5不开启,调低加大保护力度但可能降低索引效果": "保護清輔音和呼吸聲,防止電音撕裂等artifact,拉滿0.5不開啟,調低加大保護力度但可能降低索引效果",
|
30 |
+
"F0曲线文件, 可选, 一行一个音高, 代替默认F0及升降调": "F0曲線檔案,可選,一行一個音高,代替預設的F0及升降調",
|
31 |
+
"转换": "轉換",
|
32 |
+
"输出信息": "輸出訊息",
|
33 |
+
"输出音频(右下角三个点,点了可以下载)": "輸出音頻(右下角三個點,點了可以下載)",
|
34 |
+
"批量转换, 输入待转换音频文件夹, 或上传多个音频文件, 在指定文件夹(默认opt)下输出转换的音频. ": "批量轉換,輸入待轉換音頻資料夾,或上傳多個音頻檔案,在指定資料夾(默認opt)下輸出轉換的音頻。",
|
35 |
+
"指定输出文件夹": "指定輸出資料夾",
|
36 |
+
"输入待处理音频文件夹路径(去文件管理器地址栏拷就行了)": "輸入待處理音頻資料夾路徑(去檔案管理器地址欄拷貝即可)",
|
37 |
+
"也可批量输入音频文件, 二选一, 优先读文件夹": "也可批量輸入音頻檔案,二選一,優先讀資料夾",
|
38 |
+
"导出文件格式": "導出檔格式",
|
39 |
+
"伴奏人声分离&去混响&去回声": "伴奏人聲分離&去混響&去回聲",
|
40 |
+
"人声伴奏分离批量处理, 使用UVR5模型。 <br>合格的文件夹路径格式举例: E:\\codes\\py39\\vits_vc_gpu\\白鹭霜华测试样例(去文件管理器地址栏拷就行了)。 <br>模型分为三类: <br>1、保留人声:不带和声的音频选这个,对主人声保留比HP5更好。��置HP2和HP3两个模型,HP3可能轻微漏伴奏但对主人声保留比HP2稍微好一丁点; <br>2、仅保留主人声:带和声的音频选这个,对主人声可能有削弱。内置HP5一个模型; <br> 3、去混响、去延迟模型(by FoxJoy):<br> (1)MDX-Net(onnx_dereverb):对于双通道混响是最好的选择,不能去除单通道混响;<br> (234)DeEcho:去除延迟效果。Aggressive比Normal去除得更彻底,DeReverb额外去除混响,可去除单声道混响,但是对高频重的板式混响去不干净。<br>去混响/去延迟,附:<br>1、DeEcho-DeReverb模型的耗时是另外2个DeEcho模型的接近2倍;<br>2、MDX-Net-Dereverb模型挺慢的;<br>3、个人推荐的最干净的配置是先MDX-Net再DeEcho-Aggressive。": "使用UVR5模型進行人聲伴奏分離的批次處理。<br>有效資料夾路徑格式的例子:D:\\path\\to\\input\\folder(從檔案管理員地址欄複製)。<br>模型分為三類:<br>1. 保留人聲:選擇這個選項適用於沒有和聲的音訊。它比HP5更好地保留了人聲。它包括兩個內建模型:HP2和HP3。HP3可能輕微漏出伴奏,但比HP2更好地保留了人聲;<br>2. 僅保留主人聲:選擇這個選項適用於有和聲的音訊。它可能會削弱主人聲。它包括一個內建模型:HP5。<br>3. 消除混響和延遲模型(由FoxJoy提供):<br> (1) MDX-Net:對於立體聲混響的移除是最好的選擇,但不能移除單聲道混響;<br> (234) DeEcho:移除延遲效果。Aggressive模式比Normal模式移除得更徹底。DeReverb另外移除混響,可以移除單聲道混響,但對於高頻重的板式混響移除不乾淨。<br>消除混響/延遲注意事項:<br>1. DeEcho-DeReverb模型的處理時間是其他兩個DeEcho模型的近兩倍;<br>2. MDX-Net-Dereverb模型相當慢;<br>3. 個人推薦的最乾淨配置是先使用MDX-Net,然後使用DeEcho-Aggressive。",
|
41 |
+
"输入待处理音频文件夹路径": "輸入待處理音頻資料夾路徑",
|
42 |
+
"模型": "模型",
|
43 |
+
"指定输出主人声文件夹": "指定输出主人声文件夹",
|
44 |
+
"指定输出非主人声文件夹": "指定输出非主人声文件夹",
|
45 |
+
"训练": "訓練",
|
46 |
+
"step1: 填写实验配置. 实验数据放在logs下, 每个实验一个文件夹, 需手工输入实验名路径, 内含实验配置, 日志, 训练得到的模型文件. ": "step1:填寫實驗配置。實驗數據放在logs下,每個實驗一個資料夾,需手動輸入實驗名路徑,內含實驗配置、日誌、訓練得到的模型檔案。",
|
47 |
+
"输入实验名": "輸入實驗名稱",
|
48 |
+
"目标采样率": "目標取樣率",
|
49 |
+
"模型是否带音高指导(唱歌一定要, 语音可以不要)": "模型是否帶音高指導(唱歌一定要,語音可以不要)",
|
50 |
+
"版本": "版本",
|
51 |
+
"提取音高和处理数据使用的CPU进程数": "提取音高和處理數據使用的CPU進程數",
|
52 |
+
"step2a: 自动遍历训练文件夹下所有可解码成音频的文件并进行切片归一化, 在实验目录下生成2个wav文件夹; 暂时只支持单人训练. ": "step2a:自動遍歷訓練資料夾下所有可解碼成音頻的檔案並進行切片歸一化,在實驗目錄下生成2個wav資料夾;暫時只支援單人訓練。",
|
53 |
+
"输入训练文件夹路径": "輸入訓練檔案夾路徑",
|
54 |
+
"请指定说话人id": "請指定說話人id",
|
55 |
+
"处理数据": "處理資料",
|
56 |
+
"step2b: 使用CPU提取音高(如果模型带音高), 使用GPU提取特征(选择卡号)": "步驟2b: 使用CPU提取音高(如果模型帶音高), 使用GPU提取特徵(選擇卡號)",
|
57 |
+
"以-分隔输入使用的卡号, 例如 0-1-2 使用卡0和卡1和卡2": "以-分隔輸入使用的卡號, 例如 0-1-2 使用卡0和卡1和卡2",
|
58 |
+
"显卡信息": "顯示卡資訊",
|
59 |
+
"选择音高提取算法:输入歌声可用pm提速,高质量语音但CPU差可用dio提速,harvest质量更好但慢": "選擇音高提取算法:輸入歌聲可用pm提速,高品質語音但CPU差可用dio提速,harvest品質更好但較慢",
|
60 |
+
"特征提取": "特徵提取",
|
61 |
+
"step3: 填写训练设置, 开始训练模型和索引": "步驟3: 填寫訓練設定, 開始訓練模型和索引",
|
62 |
+
"保存频率save_every_epoch": "保存頻率save_every_epoch",
|
63 |
+
"总训练轮数total_epoch": "總訓練輪數total_epoch",
|
64 |
+
"每张显卡的batch_size": "每张显卡的batch_size",
|
65 |
+
"是否仅保存最新的ckpt文件以节省硬盘空间": "是否僅保存最新的ckpt檔案以節省硬碟空間",
|
66 |
+
"否": "否",
|
67 |
+
"是否缓存所有训练集至显存. 10min以下小数据可缓存以加速训练, 大数据缓存会炸显存也加不了多少速": "是否緩存所有訓練集至 VRAM。小於10分鐘的小數據可緩存以加速訓練,大數據緩存會爆 VRAM 也加不了多少速度",
|
68 |
+
"是否在每次保存时间点将最终小模型保存至weights文件夹": "是否在每次保存時間點將最終小模型保存至weights檔夾",
|
69 |
+
"加载预训练��模G路径": "加載預訓練底模G路徑",
|
70 |
+
"加载预训练底模D路径": "加載預訓練底模D路徑",
|
71 |
+
"训练模型": "訓練模型",
|
72 |
+
"训练特征索引": "訓練特徵索引",
|
73 |
+
"一键训练": "一鍵訓練",
|
74 |
+
"ckpt处理": "ckpt處理",
|
75 |
+
"模型融合, 可用于测试音色融合": "模型融合,可用於測試音色融合",
|
76 |
+
"A模型路径": "A模型路徑",
|
77 |
+
"B模型路径": "B模型路徑",
|
78 |
+
"A模型权重": "A模型權重",
|
79 |
+
"模型是否带音高指导": "模型是否帶音高指導",
|
80 |
+
"要置入的模型信息": "要置入的模型資訊",
|
81 |
+
"保存的模型名不带后缀": "儲存的模型名不帶副檔名",
|
82 |
+
"模型版本型号": "模型版本型號",
|
83 |
+
"融合": "融合",
|
84 |
+
"修改模型信息(仅支持weights文件夹下提取的小模型文件)": "修改模型資訊(僅支援weights資料夾下提取的小模型檔案)",
|
85 |
+
"模型路径": "模型路徑",
|
86 |
+
"要改的模型信息": "要改的模型資訊",
|
87 |
+
"保存的文件名, 默认空为和源文件同名": "儲存的檔案名,預設空為與來源檔案同名",
|
88 |
+
"修改": "修改",
|
89 |
+
"查看模型信息(仅支持weights文件夹下提取的小模型文件)": "查看模型資訊(僅支援weights資料夾下提取的小模型檔案)",
|
90 |
+
"查看": "查看",
|
91 |
+
"模型提取(输入logs文件夹下大文件模型路径),适用于训一半不想训了模型没有自动提取保存小文件模型,或者想测试中间模型的情况": "模型提取(輸入logs資料夾下大檔案模型路徑),適用於訓一半不想訓了模型沒有自動提取儲存小檔案模型,或者想測試中間模型的情況",
|
92 |
+
"保存名": "儲存名",
|
93 |
+
"模型是否带音高指导,1是0否": "模型是否帶音高指導,1是0否",
|
94 |
+
"提取": "提取",
|
95 |
+
"Onnx导出": "Onnx导出",
|
96 |
+
"RVC模型路径": "RVC模型路径",
|
97 |
+
"Onnx输出路径": "Onnx输出路径",
|
98 |
+
"导出Onnx模型": "导出Onnx模型",
|
99 |
+
"常见问题解答": "常見問題解答",
|
100 |
+
"招募音高曲线前端编辑器": "招募音高曲線前端編輯器",
|
101 |
+
"加开发群联系我xxxxx": "加開發群聯繫我xxxxx",
|
102 |
+
"点击查看交流、问题反馈群号": "點擊查看交流、問題反饋群號",
|
103 |
+
"xxxxx": "xxxxx",
|
104 |
+
"加载模型": "載入模型",
|
105 |
+
"Hubert模型": "Hubert 模型",
|
106 |
+
"选择.pth文件": "選擇 .pth 檔案",
|
107 |
+
"选择.index文件": "選擇 .index 檔案",
|
108 |
+
"选择.npy文件": "選擇 .npy 檔案",
|
109 |
+
"输入设备": "輸入設備",
|
110 |
+
"输出设备": "輸出設備",
|
111 |
+
"音频设备(请使用同种类驱动)": "音訊設備 (請使用同種類驅動)",
|
112 |
+
"响应阈值": "響應閾值",
|
113 |
+
"音调设置": "音調設定",
|
114 |
+
"Index Rate": "Index Rate",
|
115 |
+
"常规设置": "一般設定",
|
116 |
+
"采样长度": "取樣長度",
|
117 |
+
"淡入淡出长度": "淡入淡出長度",
|
118 |
+
"额外推理时长": "額外推理時長",
|
119 |
+
"输入降噪": "輸入降噪",
|
120 |
+
"输出降噪": "輸出降噪",
|
121 |
+
"性能设置": "效能設定",
|
122 |
+
"开始音频转换": "開始音訊轉換",
|
123 |
+
"停止音频转换": "停止音訊轉換",
|
124 |
+
"推理时间(ms):": "推理時間(ms):",
|
125 |
+
"请选择pth文件": "请选择pth文件",
|
126 |
+
"请选择index文件": "请选择index文件",
|
127 |
+
"hubert模型路径不可包含中文": "hubert模型路径不可包含中文",
|
128 |
+
"pth文件路径不可包含中文": "pth文件路径不可包含中文",
|
129 |
+
"index文件路径不可包含中文": "index文件路径不可包含中文",
|
130 |
+
"音高算法": "音高算法",
|
131 |
+
"harvest进程数": "harvest进程数"
|
132 |
+
}
|
lib/infer_pack/attentions.py
ADDED
@@ -0,0 +1,417 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import copy
|
2 |
+
import math
|
3 |
+
import numpy as np
|
4 |
+
import torch
|
5 |
+
from torch import nn
|
6 |
+
from torch.nn import functional as F
|
7 |
+
|
8 |
+
from lib.infer_pack import commons
|
9 |
+
from lib.infer_pack import modules
|
10 |
+
from lib.infer_pack.modules import LayerNorm
|
11 |
+
|
12 |
+
|
13 |
+
class Encoder(nn.Module):
|
14 |
+
def __init__(
|
15 |
+
self,
|
16 |
+
hidden_channels,
|
17 |
+
filter_channels,
|
18 |
+
n_heads,
|
19 |
+
n_layers,
|
20 |
+
kernel_size=1,
|
21 |
+
p_dropout=0.0,
|
22 |
+
window_size=10,
|
23 |
+
**kwargs
|
24 |
+
):
|
25 |
+
super().__init__()
|
26 |
+
self.hidden_channels = hidden_channels
|
27 |
+
self.filter_channels = filter_channels
|
28 |
+
self.n_heads = n_heads
|
29 |
+
self.n_layers = n_layers
|
30 |
+
self.kernel_size = kernel_size
|
31 |
+
self.p_dropout = p_dropout
|
32 |
+
self.window_size = window_size
|
33 |
+
|
34 |
+
self.drop = nn.Dropout(p_dropout)
|
35 |
+
self.attn_layers = nn.ModuleList()
|
36 |
+
self.norm_layers_1 = nn.ModuleList()
|
37 |
+
self.ffn_layers = nn.ModuleList()
|
38 |
+
self.norm_layers_2 = nn.ModuleList()
|
39 |
+
for i in range(self.n_layers):
|
40 |
+
self.attn_layers.append(
|
41 |
+
MultiHeadAttention(
|
42 |
+
hidden_channels,
|
43 |
+
hidden_channels,
|
44 |
+
n_heads,
|
45 |
+
p_dropout=p_dropout,
|
46 |
+
window_size=window_size,
|
47 |
+
)
|
48 |
+
)
|
49 |
+
self.norm_layers_1.append(LayerNorm(hidden_channels))
|
50 |
+
self.ffn_layers.append(
|
51 |
+
FFN(
|
52 |
+
hidden_channels,
|
53 |
+
hidden_channels,
|
54 |
+
filter_channels,
|
55 |
+
kernel_size,
|
56 |
+
p_dropout=p_dropout,
|
57 |
+
)
|
58 |
+
)
|
59 |
+
self.norm_layers_2.append(LayerNorm(hidden_channels))
|
60 |
+
|
61 |
+
def forward(self, x, x_mask):
|
62 |
+
attn_mask = x_mask.unsqueeze(2) * x_mask.unsqueeze(-1)
|
63 |
+
x = x * x_mask
|
64 |
+
for i in range(self.n_layers):
|
65 |
+
y = self.attn_layers[i](x, x, attn_mask)
|
66 |
+
y = self.drop(y)
|
67 |
+
x = self.norm_layers_1[i](x + y)
|
68 |
+
|
69 |
+
y = self.ffn_layers[i](x, x_mask)
|
70 |
+
y = self.drop(y)
|
71 |
+
x = self.norm_layers_2[i](x + y)
|
72 |
+
x = x * x_mask
|
73 |
+
return x
|
74 |
+
|
75 |
+
|
76 |
+
class Decoder(nn.Module):
|
77 |
+
def __init__(
|
78 |
+
self,
|
79 |
+
hidden_channels,
|
80 |
+
filter_channels,
|
81 |
+
n_heads,
|
82 |
+
n_layers,
|
83 |
+
kernel_size=1,
|
84 |
+
p_dropout=0.0,
|
85 |
+
proximal_bias=False,
|
86 |
+
proximal_init=True,
|
87 |
+
**kwargs
|
88 |
+
):
|
89 |
+
super().__init__()
|
90 |
+
self.hidden_channels = hidden_channels
|
91 |
+
self.filter_channels = filter_channels
|
92 |
+
self.n_heads = n_heads
|
93 |
+
self.n_layers = n_layers
|
94 |
+
self.kernel_size = kernel_size
|
95 |
+
self.p_dropout = p_dropout
|
96 |
+
self.proximal_bias = proximal_bias
|
97 |
+
self.proximal_init = proximal_init
|
98 |
+
|
99 |
+
self.drop = nn.Dropout(p_dropout)
|
100 |
+
self.self_attn_layers = nn.ModuleList()
|
101 |
+
self.norm_layers_0 = nn.ModuleList()
|
102 |
+
self.encdec_attn_layers = nn.ModuleList()
|
103 |
+
self.norm_layers_1 = nn.ModuleList()
|
104 |
+
self.ffn_layers = nn.ModuleList()
|
105 |
+
self.norm_layers_2 = nn.ModuleList()
|
106 |
+
for i in range(self.n_layers):
|
107 |
+
self.self_attn_layers.append(
|
108 |
+
MultiHeadAttention(
|
109 |
+
hidden_channels,
|
110 |
+
hidden_channels,
|
111 |
+
n_heads,
|
112 |
+
p_dropout=p_dropout,
|
113 |
+
proximal_bias=proximal_bias,
|
114 |
+
proximal_init=proximal_init,
|
115 |
+
)
|
116 |
+
)
|
117 |
+
self.norm_layers_0.append(LayerNorm(hidden_channels))
|
118 |
+
self.encdec_attn_layers.append(
|
119 |
+
MultiHeadAttention(
|
120 |
+
hidden_channels, hidden_channels, n_heads, p_dropout=p_dropout
|
121 |
+
)
|
122 |
+
)
|
123 |
+
self.norm_layers_1.append(LayerNorm(hidden_channels))
|
124 |
+
self.ffn_layers.append(
|
125 |
+
FFN(
|
126 |
+
hidden_channels,
|
127 |
+
hidden_channels,
|
128 |
+
filter_channels,
|
129 |
+
kernel_size,
|
130 |
+
p_dropout=p_dropout,
|
131 |
+
causal=True,
|
132 |
+
)
|
133 |
+
)
|
134 |
+
self.norm_layers_2.append(LayerNorm(hidden_channels))
|
135 |
+
|
136 |
+
def forward(self, x, x_mask, h, h_mask):
|
137 |
+
"""
|
138 |
+
x: decoder input
|
139 |
+
h: encoder output
|
140 |
+
"""
|
141 |
+
self_attn_mask = commons.subsequent_mask(x_mask.size(2)).to(
|
142 |
+
device=x.device, dtype=x.dtype
|
143 |
+
)
|
144 |
+
encdec_attn_mask = h_mask.unsqueeze(2) * x_mask.unsqueeze(-1)
|
145 |
+
x = x * x_mask
|
146 |
+
for i in range(self.n_layers):
|
147 |
+
y = self.self_attn_layers[i](x, x, self_attn_mask)
|
148 |
+
y = self.drop(y)
|
149 |
+
x = self.norm_layers_0[i](x + y)
|
150 |
+
|
151 |
+
y = self.encdec_attn_layers[i](x, h, encdec_attn_mask)
|
152 |
+
y = self.drop(y)
|
153 |
+
x = self.norm_layers_1[i](x + y)
|
154 |
+
|
155 |
+
y = self.ffn_layers[i](x, x_mask)
|
156 |
+
y = self.drop(y)
|
157 |
+
x = self.norm_layers_2[i](x + y)
|
158 |
+
x = x * x_mask
|
159 |
+
return x
|
160 |
+
|
161 |
+
|
162 |
+
class MultiHeadAttention(nn.Module):
|
163 |
+
def __init__(
|
164 |
+
self,
|
165 |
+
channels,
|
166 |
+
out_channels,
|
167 |
+
n_heads,
|
168 |
+
p_dropout=0.0,
|
169 |
+
window_size=None,
|
170 |
+
heads_share=True,
|
171 |
+
block_length=None,
|
172 |
+
proximal_bias=False,
|
173 |
+
proximal_init=False,
|
174 |
+
):
|
175 |
+
super().__init__()
|
176 |
+
assert channels % n_heads == 0
|
177 |
+
|
178 |
+
self.channels = channels
|
179 |
+
self.out_channels = out_channels
|
180 |
+
self.n_heads = n_heads
|
181 |
+
self.p_dropout = p_dropout
|
182 |
+
self.window_size = window_size
|
183 |
+
self.heads_share = heads_share
|
184 |
+
self.block_length = block_length
|
185 |
+
self.proximal_bias = proximal_bias
|
186 |
+
self.proximal_init = proximal_init
|
187 |
+
self.attn = None
|
188 |
+
|
189 |
+
self.k_channels = channels // n_heads
|
190 |
+
self.conv_q = nn.Conv1d(channels, channels, 1)
|
191 |
+
self.conv_k = nn.Conv1d(channels, channels, 1)
|
192 |
+
self.conv_v = nn.Conv1d(channels, channels, 1)
|
193 |
+
self.conv_o = nn.Conv1d(channels, out_channels, 1)
|
194 |
+
self.drop = nn.Dropout(p_dropout)
|
195 |
+
|
196 |
+
if window_size is not None:
|
197 |
+
n_heads_rel = 1 if heads_share else n_heads
|
198 |
+
rel_stddev = self.k_channels**-0.5
|
199 |
+
self.emb_rel_k = nn.Parameter(
|
200 |
+
torch.randn(n_heads_rel, window_size * 2 + 1, self.k_channels)
|
201 |
+
* rel_stddev
|
202 |
+
)
|
203 |
+
self.emb_rel_v = nn.Parameter(
|
204 |
+
torch.randn(n_heads_rel, window_size * 2 + 1, self.k_channels)
|
205 |
+
* rel_stddev
|
206 |
+
)
|
207 |
+
|
208 |
+
nn.init.xavier_uniform_(self.conv_q.weight)
|
209 |
+
nn.init.xavier_uniform_(self.conv_k.weight)
|
210 |
+
nn.init.xavier_uniform_(self.conv_v.weight)
|
211 |
+
if proximal_init:
|
212 |
+
with torch.no_grad():
|
213 |
+
self.conv_k.weight.copy_(self.conv_q.weight)
|
214 |
+
self.conv_k.bias.copy_(self.conv_q.bias)
|
215 |
+
|
216 |
+
def forward(self, x, c, attn_mask=None):
|
217 |
+
q = self.conv_q(x)
|
218 |
+
k = self.conv_k(c)
|
219 |
+
v = self.conv_v(c)
|
220 |
+
|
221 |
+
x, self.attn = self.attention(q, k, v, mask=attn_mask)
|
222 |
+
|
223 |
+
x = self.conv_o(x)
|
224 |
+
return x
|
225 |
+
|
226 |
+
def attention(self, query, key, value, mask=None):
|
227 |
+
# reshape [b, d, t] -> [b, n_h, t, d_k]
|
228 |
+
b, d, t_s, t_t = (*key.size(), query.size(2))
|
229 |
+
query = query.view(b, self.n_heads, self.k_channels, t_t).transpose(2, 3)
|
230 |
+
key = key.view(b, self.n_heads, self.k_channels, t_s).transpose(2, 3)
|
231 |
+
value = value.view(b, self.n_heads, self.k_channels, t_s).transpose(2, 3)
|
232 |
+
|
233 |
+
scores = torch.matmul(query / math.sqrt(self.k_channels), key.transpose(-2, -1))
|
234 |
+
if self.window_size is not None:
|
235 |
+
assert (
|
236 |
+
t_s == t_t
|
237 |
+
), "Relative attention is only available for self-attention."
|
238 |
+
key_relative_embeddings = self._get_relative_embeddings(self.emb_rel_k, t_s)
|
239 |
+
rel_logits = self._matmul_with_relative_keys(
|
240 |
+
query / math.sqrt(self.k_channels), key_relative_embeddings
|
241 |
+
)
|
242 |
+
scores_local = self._relative_position_to_absolute_position(rel_logits)
|
243 |
+
scores = scores + scores_local
|
244 |
+
if self.proximal_bias:
|
245 |
+
assert t_s == t_t, "Proximal bias is only available for self-attention."
|
246 |
+
scores = scores + self._attention_bias_proximal(t_s).to(
|
247 |
+
device=scores.device, dtype=scores.dtype
|
248 |
+
)
|
249 |
+
if mask is not None:
|
250 |
+
scores = scores.masked_fill(mask == 0, -1e4)
|
251 |
+
if self.block_length is not None:
|
252 |
+
assert (
|
253 |
+
t_s == t_t
|
254 |
+
), "Local attention is only available for self-attention."
|
255 |
+
block_mask = (
|
256 |
+
torch.ones_like(scores)
|
257 |
+
.triu(-self.block_length)
|
258 |
+
.tril(self.block_length)
|
259 |
+
)
|
260 |
+
scores = scores.masked_fill(block_mask == 0, -1e4)
|
261 |
+
p_attn = F.softmax(scores, dim=-1) # [b, n_h, t_t, t_s]
|
262 |
+
p_attn = self.drop(p_attn)
|
263 |
+
output = torch.matmul(p_attn, value)
|
264 |
+
if self.window_size is not None:
|
265 |
+
relative_weights = self._absolute_position_to_relative_position(p_attn)
|
266 |
+
value_relative_embeddings = self._get_relative_embeddings(
|
267 |
+
self.emb_rel_v, t_s
|
268 |
+
)
|
269 |
+
output = output + self._matmul_with_relative_values(
|
270 |
+
relative_weights, value_relative_embeddings
|
271 |
+
)
|
272 |
+
output = (
|
273 |
+
output.transpose(2, 3).contiguous().view(b, d, t_t)
|
274 |
+
) # [b, n_h, t_t, d_k] -> [b, d, t_t]
|
275 |
+
return output, p_attn
|
276 |
+
|
277 |
+
def _matmul_with_relative_values(self, x, y):
|
278 |
+
"""
|
279 |
+
x: [b, h, l, m]
|
280 |
+
y: [h or 1, m, d]
|
281 |
+
ret: [b, h, l, d]
|
282 |
+
"""
|
283 |
+
ret = torch.matmul(x, y.unsqueeze(0))
|
284 |
+
return ret
|
285 |
+
|
286 |
+
def _matmul_with_relative_keys(self, x, y):
|
287 |
+
"""
|
288 |
+
x: [b, h, l, d]
|
289 |
+
y: [h or 1, m, d]
|
290 |
+
ret: [b, h, l, m]
|
291 |
+
"""
|
292 |
+
ret = torch.matmul(x, y.unsqueeze(0).transpose(-2, -1))
|
293 |
+
return ret
|
294 |
+
|
295 |
+
def _get_relative_embeddings(self, relative_embeddings, length):
|
296 |
+
max_relative_position = 2 * self.window_size + 1
|
297 |
+
# Pad first before slice to avoid using cond ops.
|
298 |
+
pad_length = max(length - (self.window_size + 1), 0)
|
299 |
+
slice_start_position = max((self.window_size + 1) - length, 0)
|
300 |
+
slice_end_position = slice_start_position + 2 * length - 1
|
301 |
+
if pad_length > 0:
|
302 |
+
padded_relative_embeddings = F.pad(
|
303 |
+
relative_embeddings,
|
304 |
+
commons.convert_pad_shape([[0, 0], [pad_length, pad_length], [0, 0]]),
|
305 |
+
)
|
306 |
+
else:
|
307 |
+
padded_relative_embeddings = relative_embeddings
|
308 |
+
used_relative_embeddings = padded_relative_embeddings[
|
309 |
+
:, slice_start_position:slice_end_position
|
310 |
+
]
|
311 |
+
return used_relative_embeddings
|
312 |
+
|
313 |
+
def _relative_position_to_absolute_position(self, x):
|
314 |
+
"""
|
315 |
+
x: [b, h, l, 2*l-1]
|
316 |
+
ret: [b, h, l, l]
|
317 |
+
"""
|
318 |
+
batch, heads, length, _ = x.size()
|
319 |
+
# Concat columns of pad to shift from relative to absolute indexing.
|
320 |
+
x = F.pad(x, commons.convert_pad_shape([[0, 0], [0, 0], [0, 0], [0, 1]]))
|
321 |
+
|
322 |
+
# Concat extra elements so to add up to shape (len+1, 2*len-1).
|
323 |
+
x_flat = x.view([batch, heads, length * 2 * length])
|
324 |
+
x_flat = F.pad(
|
325 |
+
x_flat, commons.convert_pad_shape([[0, 0], [0, 0], [0, length - 1]])
|
326 |
+
)
|
327 |
+
|
328 |
+
# Reshape and slice out the padded elements.
|
329 |
+
x_final = x_flat.view([batch, heads, length + 1, 2 * length - 1])[
|
330 |
+
:, :, :length, length - 1 :
|
331 |
+
]
|
332 |
+
return x_final
|
333 |
+
|
334 |
+
def _absolute_position_to_relative_position(self, x):
|
335 |
+
"""
|
336 |
+
x: [b, h, l, l]
|
337 |
+
ret: [b, h, l, 2*l-1]
|
338 |
+
"""
|
339 |
+
batch, heads, length, _ = x.size()
|
340 |
+
# padd along column
|
341 |
+
x = F.pad(
|
342 |
+
x, commons.convert_pad_shape([[0, 0], [0, 0], [0, 0], [0, length - 1]])
|
343 |
+
)
|
344 |
+
x_flat = x.view([batch, heads, length**2 + length * (length - 1)])
|
345 |
+
# add 0's in the beginning that will skew the elements after reshape
|
346 |
+
x_flat = F.pad(x_flat, commons.convert_pad_shape([[0, 0], [0, 0], [length, 0]]))
|
347 |
+
x_final = x_flat.view([batch, heads, length, 2 * length])[:, :, :, 1:]
|
348 |
+
return x_final
|
349 |
+
|
350 |
+
def _attention_bias_proximal(self, length):
|
351 |
+
"""Bias for self-attention to encourage attention to close positions.
|
352 |
+
Args:
|
353 |
+
length: an integer scalar.
|
354 |
+
Returns:
|
355 |
+
a Tensor with shape [1, 1, length, length]
|
356 |
+
"""
|
357 |
+
r = torch.arange(length, dtype=torch.float32)
|
358 |
+
diff = torch.unsqueeze(r, 0) - torch.unsqueeze(r, 1)
|
359 |
+
return torch.unsqueeze(torch.unsqueeze(-torch.log1p(torch.abs(diff)), 0), 0)
|
360 |
+
|
361 |
+
|
362 |
+
class FFN(nn.Module):
|
363 |
+
def __init__(
|
364 |
+
self,
|
365 |
+
in_channels,
|
366 |
+
out_channels,
|
367 |
+
filter_channels,
|
368 |
+
kernel_size,
|
369 |
+
p_dropout=0.0,
|
370 |
+
activation=None,
|
371 |
+
causal=False,
|
372 |
+
):
|
373 |
+
super().__init__()
|
374 |
+
self.in_channels = in_channels
|
375 |
+
self.out_channels = out_channels
|
376 |
+
self.filter_channels = filter_channels
|
377 |
+
self.kernel_size = kernel_size
|
378 |
+
self.p_dropout = p_dropout
|
379 |
+
self.activation = activation
|
380 |
+
self.causal = causal
|
381 |
+
|
382 |
+
if causal:
|
383 |
+
self.padding = self._causal_padding
|
384 |
+
else:
|
385 |
+
self.padding = self._same_padding
|
386 |
+
|
387 |
+
self.conv_1 = nn.Conv1d(in_channels, filter_channels, kernel_size)
|
388 |
+
self.conv_2 = nn.Conv1d(filter_channels, out_channels, kernel_size)
|
389 |
+
self.drop = nn.Dropout(p_dropout)
|
390 |
+
|
391 |
+
def forward(self, x, x_mask):
|
392 |
+
x = self.conv_1(self.padding(x * x_mask))
|
393 |
+
if self.activation == "gelu":
|
394 |
+
x = x * torch.sigmoid(1.702 * x)
|
395 |
+
else:
|
396 |
+
x = torch.relu(x)
|
397 |
+
x = self.drop(x)
|
398 |
+
x = self.conv_2(self.padding(x * x_mask))
|
399 |
+
return x * x_mask
|
400 |
+
|
401 |
+
def _causal_padding(self, x):
|
402 |
+
if self.kernel_size == 1:
|
403 |
+
return x
|
404 |
+
pad_l = self.kernel_size - 1
|
405 |
+
pad_r = 0
|
406 |
+
padding = [[0, 0], [0, 0], [pad_l, pad_r]]
|
407 |
+
x = F.pad(x, commons.convert_pad_shape(padding))
|
408 |
+
return x
|
409 |
+
|
410 |
+
def _same_padding(self, x):
|
411 |
+
if self.kernel_size == 1:
|
412 |
+
return x
|
413 |
+
pad_l = (self.kernel_size - 1) // 2
|
414 |
+
pad_r = self.kernel_size // 2
|
415 |
+
padding = [[0, 0], [0, 0], [pad_l, pad_r]]
|
416 |
+
x = F.pad(x, commons.convert_pad_shape(padding))
|
417 |
+
return x
|
lib/infer_pack/commons.py
ADDED
@@ -0,0 +1,166 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import math
|
2 |
+
import numpy as np
|
3 |
+
import torch
|
4 |
+
from torch import nn
|
5 |
+
from torch.nn import functional as F
|
6 |
+
|
7 |
+
|
8 |
+
def init_weights(m, mean=0.0, std=0.01):
|
9 |
+
classname = m.__class__.__name__
|
10 |
+
if classname.find("Conv") != -1:
|
11 |
+
m.weight.data.normal_(mean, std)
|
12 |
+
|
13 |
+
|
14 |
+
def get_padding(kernel_size, dilation=1):
|
15 |
+
return int((kernel_size * dilation - dilation) / 2)
|
16 |
+
|
17 |
+
|
18 |
+
def convert_pad_shape(pad_shape):
|
19 |
+
l = pad_shape[::-1]
|
20 |
+
pad_shape = [item for sublist in l for item in sublist]
|
21 |
+
return pad_shape
|
22 |
+
|
23 |
+
|
24 |
+
def kl_divergence(m_p, logs_p, m_q, logs_q):
|
25 |
+
"""KL(P||Q)"""
|
26 |
+
kl = (logs_q - logs_p) - 0.5
|
27 |
+
kl += (
|
28 |
+
0.5 * (torch.exp(2.0 * logs_p) + ((m_p - m_q) ** 2)) * torch.exp(-2.0 * logs_q)
|
29 |
+
)
|
30 |
+
return kl
|
31 |
+
|
32 |
+
|
33 |
+
def rand_gumbel(shape):
|
34 |
+
"""Sample from the Gumbel distribution, protect from overflows."""
|
35 |
+
uniform_samples = torch.rand(shape) * 0.99998 + 0.00001
|
36 |
+
return -torch.log(-torch.log(uniform_samples))
|
37 |
+
|
38 |
+
|
39 |
+
def rand_gumbel_like(x):
|
40 |
+
g = rand_gumbel(x.size()).to(dtype=x.dtype, device=x.device)
|
41 |
+
return g
|
42 |
+
|
43 |
+
|
44 |
+
def slice_segments(x, ids_str, segment_size=4):
|
45 |
+
ret = torch.zeros_like(x[:, :, :segment_size])
|
46 |
+
for i in range(x.size(0)):
|
47 |
+
idx_str = ids_str[i]
|
48 |
+
idx_end = idx_str + segment_size
|
49 |
+
ret[i] = x[i, :, idx_str:idx_end]
|
50 |
+
return ret
|
51 |
+
|
52 |
+
|
53 |
+
def slice_segments2(x, ids_str, segment_size=4):
|
54 |
+
ret = torch.zeros_like(x[:, :segment_size])
|
55 |
+
for i in range(x.size(0)):
|
56 |
+
idx_str = ids_str[i]
|
57 |
+
idx_end = idx_str + segment_size
|
58 |
+
ret[i] = x[i, idx_str:idx_end]
|
59 |
+
return ret
|
60 |
+
|
61 |
+
|
62 |
+
def rand_slice_segments(x, x_lengths=None, segment_size=4):
|
63 |
+
b, d, t = x.size()
|
64 |
+
if x_lengths is None:
|
65 |
+
x_lengths = t
|
66 |
+
ids_str_max = x_lengths - segment_size + 1
|
67 |
+
ids_str = (torch.rand([b]).to(device=x.device) * ids_str_max).to(dtype=torch.long)
|
68 |
+
ret = slice_segments(x, ids_str, segment_size)
|
69 |
+
return ret, ids_str
|
70 |
+
|
71 |
+
|
72 |
+
def get_timing_signal_1d(length, channels, min_timescale=1.0, max_timescale=1.0e4):
|
73 |
+
position = torch.arange(length, dtype=torch.float)
|
74 |
+
num_timescales = channels // 2
|
75 |
+
log_timescale_increment = math.log(float(max_timescale) / float(min_timescale)) / (
|
76 |
+
num_timescales - 1
|
77 |
+
)
|
78 |
+
inv_timescales = min_timescale * torch.exp(
|
79 |
+
torch.arange(num_timescales, dtype=torch.float) * -log_timescale_increment
|
80 |
+
)
|
81 |
+
scaled_time = position.unsqueeze(0) * inv_timescales.unsqueeze(1)
|
82 |
+
signal = torch.cat([torch.sin(scaled_time), torch.cos(scaled_time)], 0)
|
83 |
+
signal = F.pad(signal, [0, 0, 0, channels % 2])
|
84 |
+
signal = signal.view(1, channels, length)
|
85 |
+
return signal
|
86 |
+
|
87 |
+
|
88 |
+
def add_timing_signal_1d(x, min_timescale=1.0, max_timescale=1.0e4):
|
89 |
+
b, channels, length = x.size()
|
90 |
+
signal = get_timing_signal_1d(length, channels, min_timescale, max_timescale)
|
91 |
+
return x + signal.to(dtype=x.dtype, device=x.device)
|
92 |
+
|
93 |
+
|
94 |
+
def cat_timing_signal_1d(x, min_timescale=1.0, max_timescale=1.0e4, axis=1):
|
95 |
+
b, channels, length = x.size()
|
96 |
+
signal = get_timing_signal_1d(length, channels, min_timescale, max_timescale)
|
97 |
+
return torch.cat([x, signal.to(dtype=x.dtype, device=x.device)], axis)
|
98 |
+
|
99 |
+
|
100 |
+
def subsequent_mask(length):
|
101 |
+
mask = torch.tril(torch.ones(length, length)).unsqueeze(0).unsqueeze(0)
|
102 |
+
return mask
|
103 |
+
|
104 |
+
|
105 |
+
@torch.jit.script
|
106 |
+
def fused_add_tanh_sigmoid_multiply(input_a, input_b, n_channels):
|
107 |
+
n_channels_int = n_channels[0]
|
108 |
+
in_act = input_a + input_b
|
109 |
+
t_act = torch.tanh(in_act[:, :n_channels_int, :])
|
110 |
+
s_act = torch.sigmoid(in_act[:, n_channels_int:, :])
|
111 |
+
acts = t_act * s_act
|
112 |
+
return acts
|
113 |
+
|
114 |
+
|
115 |
+
def convert_pad_shape(pad_shape):
|
116 |
+
l = pad_shape[::-1]
|
117 |
+
pad_shape = [item for sublist in l for item in sublist]
|
118 |
+
return pad_shape
|
119 |
+
|
120 |
+
|
121 |
+
def shift_1d(x):
|
122 |
+
x = F.pad(x, convert_pad_shape([[0, 0], [0, 0], [1, 0]]))[:, :, :-1]
|
123 |
+
return x
|
124 |
+
|
125 |
+
|
126 |
+
def sequence_mask(length, max_length=None):
|
127 |
+
if max_length is None:
|
128 |
+
max_length = length.max()
|
129 |
+
x = torch.arange(max_length, dtype=length.dtype, device=length.device)
|
130 |
+
return x.unsqueeze(0) < length.unsqueeze(1)
|
131 |
+
|
132 |
+
|
133 |
+
def generate_path(duration, mask):
|
134 |
+
"""
|
135 |
+
duration: [b, 1, t_x]
|
136 |
+
mask: [b, 1, t_y, t_x]
|
137 |
+
"""
|
138 |
+
device = duration.device
|
139 |
+
|
140 |
+
b, _, t_y, t_x = mask.shape
|
141 |
+
cum_duration = torch.cumsum(duration, -1)
|
142 |
+
|
143 |
+
cum_duration_flat = cum_duration.view(b * t_x)
|
144 |
+
path = sequence_mask(cum_duration_flat, t_y).to(mask.dtype)
|
145 |
+
path = path.view(b, t_x, t_y)
|
146 |
+
path = path - F.pad(path, convert_pad_shape([[0, 0], [1, 0], [0, 0]]))[:, :-1]
|
147 |
+
path = path.unsqueeze(1).transpose(2, 3) * mask
|
148 |
+
return path
|
149 |
+
|
150 |
+
|
151 |
+
def clip_grad_value_(parameters, clip_value, norm_type=2):
|
152 |
+
if isinstance(parameters, torch.Tensor):
|
153 |
+
parameters = [parameters]
|
154 |
+
parameters = list(filter(lambda p: p.grad is not None, parameters))
|
155 |
+
norm_type = float(norm_type)
|
156 |
+
if clip_value is not None:
|
157 |
+
clip_value = float(clip_value)
|
158 |
+
|
159 |
+
total_norm = 0
|
160 |
+
for p in parameters:
|
161 |
+
param_norm = p.grad.data.norm(norm_type)
|
162 |
+
total_norm += param_norm.item() ** norm_type
|
163 |
+
if clip_value is not None:
|
164 |
+
p.grad.data.clamp_(min=-clip_value, max=clip_value)
|
165 |
+
total_norm = total_norm ** (1.0 / norm_type)
|
166 |
+
return total_norm
|
lib/infer_pack/models.py
ADDED
@@ -0,0 +1,1142 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import math, pdb, os
|
2 |
+
from time import time as ttime
|
3 |
+
import torch
|
4 |
+
from torch import nn
|
5 |
+
from torch.nn import functional as F
|
6 |
+
from lib.infer_pack import modules
|
7 |
+
from lib.infer_pack import attentions
|
8 |
+
from lib.infer_pack import commons
|
9 |
+
from lib.infer_pack.commons import init_weights, get_padding
|
10 |
+
from torch.nn import Conv1d, ConvTranspose1d, AvgPool1d, Conv2d
|
11 |
+
from torch.nn.utils import weight_norm, remove_weight_norm, spectral_norm
|
12 |
+
from lib.infer_pack.commons import init_weights
|
13 |
+
import numpy as np
|
14 |
+
from lib.infer_pack import commons
|
15 |
+
|
16 |
+
|
17 |
+
class TextEncoder256(nn.Module):
|
18 |
+
def __init__(
|
19 |
+
self,
|
20 |
+
out_channels,
|
21 |
+
hidden_channels,
|
22 |
+
filter_channels,
|
23 |
+
n_heads,
|
24 |
+
n_layers,
|
25 |
+
kernel_size,
|
26 |
+
p_dropout,
|
27 |
+
f0=True,
|
28 |
+
):
|
29 |
+
super().__init__()
|
30 |
+
self.out_channels = out_channels
|
31 |
+
self.hidden_channels = hidden_channels
|
32 |
+
self.filter_channels = filter_channels
|
33 |
+
self.n_heads = n_heads
|
34 |
+
self.n_layers = n_layers
|
35 |
+
self.kernel_size = kernel_size
|
36 |
+
self.p_dropout = p_dropout
|
37 |
+
self.emb_phone = nn.Linear(256, hidden_channels)
|
38 |
+
self.lrelu = nn.LeakyReLU(0.1, inplace=True)
|
39 |
+
if f0 == True:
|
40 |
+
self.emb_pitch = nn.Embedding(256, hidden_channels) # pitch 256
|
41 |
+
self.encoder = attentions.Encoder(
|
42 |
+
hidden_channels, filter_channels, n_heads, n_layers, kernel_size, p_dropout
|
43 |
+
)
|
44 |
+
self.proj = nn.Conv1d(hidden_channels, out_channels * 2, 1)
|
45 |
+
|
46 |
+
def forward(self, phone, pitch, lengths):
|
47 |
+
if pitch == None:
|
48 |
+
x = self.emb_phone(phone)
|
49 |
+
else:
|
50 |
+
x = self.emb_phone(phone) + self.emb_pitch(pitch)
|
51 |
+
x = x * math.sqrt(self.hidden_channels) # [b, t, h]
|
52 |
+
x = self.lrelu(x)
|
53 |
+
x = torch.transpose(x, 1, -1) # [b, h, t]
|
54 |
+
x_mask = torch.unsqueeze(commons.sequence_mask(lengths, x.size(2)), 1).to(
|
55 |
+
x.dtype
|
56 |
+
)
|
57 |
+
x = self.encoder(x * x_mask, x_mask)
|
58 |
+
stats = self.proj(x) * x_mask
|
59 |
+
|
60 |
+
m, logs = torch.split(stats, self.out_channels, dim=1)
|
61 |
+
return m, logs, x_mask
|
62 |
+
|
63 |
+
|
64 |
+
class TextEncoder768(nn.Module):
|
65 |
+
def __init__(
|
66 |
+
self,
|
67 |
+
out_channels,
|
68 |
+
hidden_channels,
|
69 |
+
filter_channels,
|
70 |
+
n_heads,
|
71 |
+
n_layers,
|
72 |
+
kernel_size,
|
73 |
+
p_dropout,
|
74 |
+
f0=True,
|
75 |
+
):
|
76 |
+
super().__init__()
|
77 |
+
self.out_channels = out_channels
|
78 |
+
self.hidden_channels = hidden_channels
|
79 |
+
self.filter_channels = filter_channels
|
80 |
+
self.n_heads = n_heads
|
81 |
+
self.n_layers = n_layers
|
82 |
+
self.kernel_size = kernel_size
|
83 |
+
self.p_dropout = p_dropout
|
84 |
+
self.emb_phone = nn.Linear(768, hidden_channels)
|
85 |
+
self.lrelu = nn.LeakyReLU(0.1, inplace=True)
|
86 |
+
if f0 == True:
|
87 |
+
self.emb_pitch = nn.Embedding(256, hidden_channels) # pitch 256
|
88 |
+
self.encoder = attentions.Encoder(
|
89 |
+
hidden_channels, filter_channels, n_heads, n_layers, kernel_size, p_dropout
|
90 |
+
)
|
91 |
+
self.proj = nn.Conv1d(hidden_channels, out_channels * 2, 1)
|
92 |
+
|
93 |
+
def forward(self, phone, pitch, lengths):
|
94 |
+
if pitch == None:
|
95 |
+
x = self.emb_phone(phone)
|
96 |
+
else:
|
97 |
+
x = self.emb_phone(phone) + self.emb_pitch(pitch)
|
98 |
+
x = x * math.sqrt(self.hidden_channels) # [b, t, h]
|
99 |
+
x = self.lrelu(x)
|
100 |
+
x = torch.transpose(x, 1, -1) # [b, h, t]
|
101 |
+
x_mask = torch.unsqueeze(commons.sequence_mask(lengths, x.size(2)), 1).to(
|
102 |
+
x.dtype
|
103 |
+
)
|
104 |
+
x = self.encoder(x * x_mask, x_mask)
|
105 |
+
stats = self.proj(x) * x_mask
|
106 |
+
|
107 |
+
m, logs = torch.split(stats, self.out_channels, dim=1)
|
108 |
+
return m, logs, x_mask
|
109 |
+
|
110 |
+
|
111 |
+
class ResidualCouplingBlock(nn.Module):
|
112 |
+
def __init__(
|
113 |
+
self,
|
114 |
+
channels,
|
115 |
+
hidden_channels,
|
116 |
+
kernel_size,
|
117 |
+
dilation_rate,
|
118 |
+
n_layers,
|
119 |
+
n_flows=4,
|
120 |
+
gin_channels=0,
|
121 |
+
):
|
122 |
+
super().__init__()
|
123 |
+
self.channels = channels
|
124 |
+
self.hidden_channels = hidden_channels
|
125 |
+
self.kernel_size = kernel_size
|
126 |
+
self.dilation_rate = dilation_rate
|
127 |
+
self.n_layers = n_layers
|
128 |
+
self.n_flows = n_flows
|
129 |
+
self.gin_channels = gin_channels
|
130 |
+
|
131 |
+
self.flows = nn.ModuleList()
|
132 |
+
for i in range(n_flows):
|
133 |
+
self.flows.append(
|
134 |
+
modules.ResidualCouplingLayer(
|
135 |
+
channels,
|
136 |
+
hidden_channels,
|
137 |
+
kernel_size,
|
138 |
+
dilation_rate,
|
139 |
+
n_layers,
|
140 |
+
gin_channels=gin_channels,
|
141 |
+
mean_only=True,
|
142 |
+
)
|
143 |
+
)
|
144 |
+
self.flows.append(modules.Flip())
|
145 |
+
|
146 |
+
def forward(self, x, x_mask, g=None, reverse=False):
|
147 |
+
if not reverse:
|
148 |
+
for flow in self.flows:
|
149 |
+
x, _ = flow(x, x_mask, g=g, reverse=reverse)
|
150 |
+
else:
|
151 |
+
for flow in reversed(self.flows):
|
152 |
+
x = flow(x, x_mask, g=g, reverse=reverse)
|
153 |
+
return x
|
154 |
+
|
155 |
+
def remove_weight_norm(self):
|
156 |
+
for i in range(self.n_flows):
|
157 |
+
self.flows[i * 2].remove_weight_norm()
|
158 |
+
|
159 |
+
|
160 |
+
class PosteriorEncoder(nn.Module):
|
161 |
+
def __init__(
|
162 |
+
self,
|
163 |
+
in_channels,
|
164 |
+
out_channels,
|
165 |
+
hidden_channels,
|
166 |
+
kernel_size,
|
167 |
+
dilation_rate,
|
168 |
+
n_layers,
|
169 |
+
gin_channels=0,
|
170 |
+
):
|
171 |
+
super().__init__()
|
172 |
+
self.in_channels = in_channels
|
173 |
+
self.out_channels = out_channels
|
174 |
+
self.hidden_channels = hidden_channels
|
175 |
+
self.kernel_size = kernel_size
|
176 |
+
self.dilation_rate = dilation_rate
|
177 |
+
self.n_layers = n_layers
|
178 |
+
self.gin_channels = gin_channels
|
179 |
+
|
180 |
+
self.pre = nn.Conv1d(in_channels, hidden_channels, 1)
|
181 |
+
self.enc = modules.WN(
|
182 |
+
hidden_channels,
|
183 |
+
kernel_size,
|
184 |
+
dilation_rate,
|
185 |
+
n_layers,
|
186 |
+
gin_channels=gin_channels,
|
187 |
+
)
|
188 |
+
self.proj = nn.Conv1d(hidden_channels, out_channels * 2, 1)
|
189 |
+
|
190 |
+
def forward(self, x, x_lengths, g=None):
|
191 |
+
x_mask = torch.unsqueeze(commons.sequence_mask(x_lengths, x.size(2)), 1).to(
|
192 |
+
x.dtype
|
193 |
+
)
|
194 |
+
x = self.pre(x) * x_mask
|
195 |
+
x = self.enc(x, x_mask, g=g)
|
196 |
+
stats = self.proj(x) * x_mask
|
197 |
+
m, logs = torch.split(stats, self.out_channels, dim=1)
|
198 |
+
z = (m + torch.randn_like(m) * torch.exp(logs)) * x_mask
|
199 |
+
return z, m, logs, x_mask
|
200 |
+
|
201 |
+
def remove_weight_norm(self):
|
202 |
+
self.enc.remove_weight_norm()
|
203 |
+
|
204 |
+
|
205 |
+
class Generator(torch.nn.Module):
|
206 |
+
def __init__(
|
207 |
+
self,
|
208 |
+
initial_channel,
|
209 |
+
resblock,
|
210 |
+
resblock_kernel_sizes,
|
211 |
+
resblock_dilation_sizes,
|
212 |
+
upsample_rates,
|
213 |
+
upsample_initial_channel,
|
214 |
+
upsample_kernel_sizes,
|
215 |
+
gin_channels=0,
|
216 |
+
):
|
217 |
+
super(Generator, self).__init__()
|
218 |
+
self.num_kernels = len(resblock_kernel_sizes)
|
219 |
+
self.num_upsamples = len(upsample_rates)
|
220 |
+
self.conv_pre = Conv1d(
|
221 |
+
initial_channel, upsample_initial_channel, 7, 1, padding=3
|
222 |
+
)
|
223 |
+
resblock = modules.ResBlock1 if resblock == "1" else modules.ResBlock2
|
224 |
+
|
225 |
+
self.ups = nn.ModuleList()
|
226 |
+
for i, (u, k) in enumerate(zip(upsample_rates, upsample_kernel_sizes)):
|
227 |
+
self.ups.append(
|
228 |
+
weight_norm(
|
229 |
+
ConvTranspose1d(
|
230 |
+
upsample_initial_channel // (2**i),
|
231 |
+
upsample_initial_channel // (2 ** (i + 1)),
|
232 |
+
k,
|
233 |
+
u,
|
234 |
+
padding=(k - u) // 2,
|
235 |
+
)
|
236 |
+
)
|
237 |
+
)
|
238 |
+
|
239 |
+
self.resblocks = nn.ModuleList()
|
240 |
+
for i in range(len(self.ups)):
|
241 |
+
ch = upsample_initial_channel // (2 ** (i + 1))
|
242 |
+
for j, (k, d) in enumerate(
|
243 |
+
zip(resblock_kernel_sizes, resblock_dilation_sizes)
|
244 |
+
):
|
245 |
+
self.resblocks.append(resblock(ch, k, d))
|
246 |
+
|
247 |
+
self.conv_post = Conv1d(ch, 1, 7, 1, padding=3, bias=False)
|
248 |
+
self.ups.apply(init_weights)
|
249 |
+
|
250 |
+
if gin_channels != 0:
|
251 |
+
self.cond = nn.Conv1d(gin_channels, upsample_initial_channel, 1)
|
252 |
+
|
253 |
+
def forward(self, x, g=None):
|
254 |
+
x = self.conv_pre(x)
|
255 |
+
if g is not None:
|
256 |
+
x = x + self.cond(g)
|
257 |
+
|
258 |
+
for i in range(self.num_upsamples):
|
259 |
+
x = F.leaky_relu(x, modules.LRELU_SLOPE)
|
260 |
+
x = self.ups[i](x)
|
261 |
+
xs = None
|
262 |
+
for j in range(self.num_kernels):
|
263 |
+
if xs is None:
|
264 |
+
xs = self.resblocks[i * self.num_kernels + j](x)
|
265 |
+
else:
|
266 |
+
xs += self.resblocks[i * self.num_kernels + j](x)
|
267 |
+
x = xs / self.num_kernels
|
268 |
+
x = F.leaky_relu(x)
|
269 |
+
x = self.conv_post(x)
|
270 |
+
x = torch.tanh(x)
|
271 |
+
|
272 |
+
return x
|
273 |
+
|
274 |
+
def remove_weight_norm(self):
|
275 |
+
for l in self.ups:
|
276 |
+
remove_weight_norm(l)
|
277 |
+
for l in self.resblocks:
|
278 |
+
l.remove_weight_norm()
|
279 |
+
|
280 |
+
|
281 |
+
class SineGen(torch.nn.Module):
|
282 |
+
"""Definition of sine generator
|
283 |
+
SineGen(samp_rate, harmonic_num = 0,
|
284 |
+
sine_amp = 0.1, noise_std = 0.003,
|
285 |
+
voiced_threshold = 0,
|
286 |
+
flag_for_pulse=False)
|
287 |
+
samp_rate: sampling rate in Hz
|
288 |
+
harmonic_num: number of harmonic overtones (default 0)
|
289 |
+
sine_amp: amplitude of sine-wavefrom (default 0.1)
|
290 |
+
noise_std: std of Gaussian noise (default 0.003)
|
291 |
+
voiced_thoreshold: F0 threshold for U/V classification (default 0)
|
292 |
+
flag_for_pulse: this SinGen is used inside PulseGen (default False)
|
293 |
+
Note: when flag_for_pulse is True, the first time step of a voiced
|
294 |
+
segment is always sin(np.pi) or cos(0)
|
295 |
+
"""
|
296 |
+
|
297 |
+
def __init__(
|
298 |
+
self,
|
299 |
+
samp_rate,
|
300 |
+
harmonic_num=0,
|
301 |
+
sine_amp=0.1,
|
302 |
+
noise_std=0.003,
|
303 |
+
voiced_threshold=0,
|
304 |
+
flag_for_pulse=False,
|
305 |
+
):
|
306 |
+
super(SineGen, self).__init__()
|
307 |
+
self.sine_amp = sine_amp
|
308 |
+
self.noise_std = noise_std
|
309 |
+
self.harmonic_num = harmonic_num
|
310 |
+
self.dim = self.harmonic_num + 1
|
311 |
+
self.sampling_rate = samp_rate
|
312 |
+
self.voiced_threshold = voiced_threshold
|
313 |
+
|
314 |
+
def _f02uv(self, f0):
|
315 |
+
# generate uv signal
|
316 |
+
uv = torch.ones_like(f0)
|
317 |
+
uv = uv * (f0 > self.voiced_threshold)
|
318 |
+
return uv
|
319 |
+
|
320 |
+
def forward(self, f0, upp):
|
321 |
+
"""sine_tensor, uv = forward(f0)
|
322 |
+
input F0: tensor(batchsize=1, length, dim=1)
|
323 |
+
f0 for unvoiced steps should be 0
|
324 |
+
output sine_tensor: tensor(batchsize=1, length, dim)
|
325 |
+
output uv: tensor(batchsize=1, length, 1)
|
326 |
+
"""
|
327 |
+
with torch.no_grad():
|
328 |
+
f0 = f0[:, None].transpose(1, 2)
|
329 |
+
f0_buf = torch.zeros(f0.shape[0], f0.shape[1], self.dim, device=f0.device)
|
330 |
+
# fundamental component
|
331 |
+
f0_buf[:, :, 0] = f0[:, :, 0]
|
332 |
+
for idx in np.arange(self.harmonic_num):
|
333 |
+
f0_buf[:, :, idx + 1] = f0_buf[:, :, 0] * (
|
334 |
+
idx + 2
|
335 |
+
) # idx + 2: the (idx+1)-th overtone, (idx+2)-th harmonic
|
336 |
+
rad_values = (f0_buf / self.sampling_rate) % 1 ###%1意味着n_har的乘积无法后处理优化
|
337 |
+
rand_ini = torch.rand(
|
338 |
+
f0_buf.shape[0], f0_buf.shape[2], device=f0_buf.device
|
339 |
+
)
|
340 |
+
rand_ini[:, 0] = 0
|
341 |
+
rad_values[:, 0, :] = rad_values[:, 0, :] + rand_ini
|
342 |
+
tmp_over_one = torch.cumsum(rad_values, 1) # % 1 #####%1意味着后面的cumsum无法再优化
|
343 |
+
tmp_over_one *= upp
|
344 |
+
tmp_over_one = F.interpolate(
|
345 |
+
tmp_over_one.transpose(2, 1),
|
346 |
+
scale_factor=upp,
|
347 |
+
mode="linear",
|
348 |
+
align_corners=True,
|
349 |
+
).transpose(2, 1)
|
350 |
+
rad_values = F.interpolate(
|
351 |
+
rad_values.transpose(2, 1), scale_factor=upp, mode="nearest"
|
352 |
+
).transpose(
|
353 |
+
2, 1
|
354 |
+
) #######
|
355 |
+
tmp_over_one %= 1
|
356 |
+
tmp_over_one_idx = (tmp_over_one[:, 1:, :] - tmp_over_one[:, :-1, :]) < 0
|
357 |
+
cumsum_shift = torch.zeros_like(rad_values)
|
358 |
+
cumsum_shift[:, 1:, :] = tmp_over_one_idx * -1.0
|
359 |
+
sine_waves = torch.sin(
|
360 |
+
torch.cumsum(rad_values + cumsum_shift, dim=1) * 2 * np.pi
|
361 |
+
)
|
362 |
+
sine_waves = sine_waves * self.sine_amp
|
363 |
+
uv = self._f02uv(f0)
|
364 |
+
uv = F.interpolate(
|
365 |
+
uv.transpose(2, 1), scale_factor=upp, mode="nearest"
|
366 |
+
).transpose(2, 1)
|
367 |
+
noise_amp = uv * self.noise_std + (1 - uv) * self.sine_amp / 3
|
368 |
+
noise = noise_amp * torch.randn_like(sine_waves)
|
369 |
+
sine_waves = sine_waves * uv + noise
|
370 |
+
return sine_waves, uv, noise
|
371 |
+
|
372 |
+
|
373 |
+
class SourceModuleHnNSF(torch.nn.Module):
|
374 |
+
"""SourceModule for hn-nsf
|
375 |
+
SourceModule(sampling_rate, harmonic_num=0, sine_amp=0.1,
|
376 |
+
add_noise_std=0.003, voiced_threshod=0)
|
377 |
+
sampling_rate: sampling_rate in Hz
|
378 |
+
harmonic_num: number of harmonic above F0 (default: 0)
|
379 |
+
sine_amp: amplitude of sine source signal (default: 0.1)
|
380 |
+
add_noise_std: std of additive Gaussian noise (default: 0.003)
|
381 |
+
note that amplitude of noise in unvoiced is decided
|
382 |
+
by sine_amp
|
383 |
+
voiced_threshold: threhold to set U/V given F0 (default: 0)
|
384 |
+
Sine_source, noise_source = SourceModuleHnNSF(F0_sampled)
|
385 |
+
F0_sampled (batchsize, length, 1)
|
386 |
+
Sine_source (batchsize, length, 1)
|
387 |
+
noise_source (batchsize, length 1)
|
388 |
+
uv (batchsize, length, 1)
|
389 |
+
"""
|
390 |
+
|
391 |
+
def __init__(
|
392 |
+
self,
|
393 |
+
sampling_rate,
|
394 |
+
harmonic_num=0,
|
395 |
+
sine_amp=0.1,
|
396 |
+
add_noise_std=0.003,
|
397 |
+
voiced_threshod=0,
|
398 |
+
is_half=True,
|
399 |
+
):
|
400 |
+
super(SourceModuleHnNSF, self).__init__()
|
401 |
+
|
402 |
+
self.sine_amp = sine_amp
|
403 |
+
self.noise_std = add_noise_std
|
404 |
+
self.is_half = is_half
|
405 |
+
# to produce sine waveforms
|
406 |
+
self.l_sin_gen = SineGen(
|
407 |
+
sampling_rate, harmonic_num, sine_amp, add_noise_std, voiced_threshod
|
408 |
+
)
|
409 |
+
|
410 |
+
# to merge source harmonics into a single excitation
|
411 |
+
self.l_linear = torch.nn.Linear(harmonic_num + 1, 1)
|
412 |
+
self.l_tanh = torch.nn.Tanh()
|
413 |
+
|
414 |
+
def forward(self, x, upp=None):
|
415 |
+
sine_wavs, uv, _ = self.l_sin_gen(x, upp)
|
416 |
+
if self.is_half:
|
417 |
+
sine_wavs = sine_wavs.half()
|
418 |
+
sine_merge = self.l_tanh(self.l_linear(sine_wavs))
|
419 |
+
return sine_merge, None, None # noise, uv
|
420 |
+
|
421 |
+
|
422 |
+
class GeneratorNSF(torch.nn.Module):
|
423 |
+
def __init__(
|
424 |
+
self,
|
425 |
+
initial_channel,
|
426 |
+
resblock,
|
427 |
+
resblock_kernel_sizes,
|
428 |
+
resblock_dilation_sizes,
|
429 |
+
upsample_rates,
|
430 |
+
upsample_initial_channel,
|
431 |
+
upsample_kernel_sizes,
|
432 |
+
gin_channels,
|
433 |
+
sr,
|
434 |
+
is_half=False,
|
435 |
+
):
|
436 |
+
super(GeneratorNSF, self).__init__()
|
437 |
+
self.num_kernels = len(resblock_kernel_sizes)
|
438 |
+
self.num_upsamples = len(upsample_rates)
|
439 |
+
|
440 |
+
self.f0_upsamp = torch.nn.Upsample(scale_factor=np.prod(upsample_rates))
|
441 |
+
self.m_source = SourceModuleHnNSF(
|
442 |
+
sampling_rate=sr, harmonic_num=0, is_half=is_half
|
443 |
+
)
|
444 |
+
self.noise_convs = nn.ModuleList()
|
445 |
+
self.conv_pre = Conv1d(
|
446 |
+
initial_channel, upsample_initial_channel, 7, 1, padding=3
|
447 |
+
)
|
448 |
+
resblock = modules.ResBlock1 if resblock == "1" else modules.ResBlock2
|
449 |
+
|
450 |
+
self.ups = nn.ModuleList()
|
451 |
+
for i, (u, k) in enumerate(zip(upsample_rates, upsample_kernel_sizes)):
|
452 |
+
c_cur = upsample_initial_channel // (2 ** (i + 1))
|
453 |
+
self.ups.append(
|
454 |
+
weight_norm(
|
455 |
+
ConvTranspose1d(
|
456 |
+
upsample_initial_channel // (2**i),
|
457 |
+
upsample_initial_channel // (2 ** (i + 1)),
|
458 |
+
k,
|
459 |
+
u,
|
460 |
+
padding=(k - u) // 2,
|
461 |
+
)
|
462 |
+
)
|
463 |
+
)
|
464 |
+
if i + 1 < len(upsample_rates):
|
465 |
+
stride_f0 = np.prod(upsample_rates[i + 1 :])
|
466 |
+
self.noise_convs.append(
|
467 |
+
Conv1d(
|
468 |
+
1,
|
469 |
+
c_cur,
|
470 |
+
kernel_size=stride_f0 * 2,
|
471 |
+
stride=stride_f0,
|
472 |
+
padding=stride_f0 // 2,
|
473 |
+
)
|
474 |
+
)
|
475 |
+
else:
|
476 |
+
self.noise_convs.append(Conv1d(1, c_cur, kernel_size=1))
|
477 |
+
|
478 |
+
self.resblocks = nn.ModuleList()
|
479 |
+
for i in range(len(self.ups)):
|
480 |
+
ch = upsample_initial_channel // (2 ** (i + 1))
|
481 |
+
for j, (k, d) in enumerate(
|
482 |
+
zip(resblock_kernel_sizes, resblock_dilation_sizes)
|
483 |
+
):
|
484 |
+
self.resblocks.append(resblock(ch, k, d))
|
485 |
+
|
486 |
+
self.conv_post = Conv1d(ch, 1, 7, 1, padding=3, bias=False)
|
487 |
+
self.ups.apply(init_weights)
|
488 |
+
|
489 |
+
if gin_channels != 0:
|
490 |
+
self.cond = nn.Conv1d(gin_channels, upsample_initial_channel, 1)
|
491 |
+
|
492 |
+
self.upp = np.prod(upsample_rates)
|
493 |
+
|
494 |
+
def forward(self, x, f0, g=None):
|
495 |
+
har_source, noi_source, uv = self.m_source(f0, self.upp)
|
496 |
+
har_source = har_source.transpose(1, 2)
|
497 |
+
x = self.conv_pre(x)
|
498 |
+
if g is not None:
|
499 |
+
x = x + self.cond(g)
|
500 |
+
|
501 |
+
for i in range(self.num_upsamples):
|
502 |
+
x = F.leaky_relu(x, modules.LRELU_SLOPE)
|
503 |
+
x = self.ups[i](x)
|
504 |
+
x_source = self.noise_convs[i](har_source)
|
505 |
+
x = x + x_source
|
506 |
+
xs = None
|
507 |
+
for j in range(self.num_kernels):
|
508 |
+
if xs is None:
|
509 |
+
xs = self.resblocks[i * self.num_kernels + j](x)
|
510 |
+
else:
|
511 |
+
xs += self.resblocks[i * self.num_kernels + j](x)
|
512 |
+
x = xs / self.num_kernels
|
513 |
+
x = F.leaky_relu(x)
|
514 |
+
x = self.conv_post(x)
|
515 |
+
x = torch.tanh(x)
|
516 |
+
return x
|
517 |
+
|
518 |
+
def remove_weight_norm(self):
|
519 |
+
for l in self.ups:
|
520 |
+
remove_weight_norm(l)
|
521 |
+
for l in self.resblocks:
|
522 |
+
l.remove_weight_norm()
|
523 |
+
|
524 |
+
|
525 |
+
sr2sr = {
|
526 |
+
"32k": 32000,
|
527 |
+
"40k": 40000,
|
528 |
+
"48k": 48000,
|
529 |
+
}
|
530 |
+
|
531 |
+
|
532 |
+
class SynthesizerTrnMs256NSFsid(nn.Module):
|
533 |
+
def __init__(
|
534 |
+
self,
|
535 |
+
spec_channels,
|
536 |
+
segment_size,
|
537 |
+
inter_channels,
|
538 |
+
hidden_channels,
|
539 |
+
filter_channels,
|
540 |
+
n_heads,
|
541 |
+
n_layers,
|
542 |
+
kernel_size,
|
543 |
+
p_dropout,
|
544 |
+
resblock,
|
545 |
+
resblock_kernel_sizes,
|
546 |
+
resblock_dilation_sizes,
|
547 |
+
upsample_rates,
|
548 |
+
upsample_initial_channel,
|
549 |
+
upsample_kernel_sizes,
|
550 |
+
spk_embed_dim,
|
551 |
+
gin_channels,
|
552 |
+
sr,
|
553 |
+
**kwargs
|
554 |
+
):
|
555 |
+
super().__init__()
|
556 |
+
if type(sr) == type("strr"):
|
557 |
+
sr = sr2sr[sr]
|
558 |
+
self.spec_channels = spec_channels
|
559 |
+
self.inter_channels = inter_channels
|
560 |
+
self.hidden_channels = hidden_channels
|
561 |
+
self.filter_channels = filter_channels
|
562 |
+
self.n_heads = n_heads
|
563 |
+
self.n_layers = n_layers
|
564 |
+
self.kernel_size = kernel_size
|
565 |
+
self.p_dropout = p_dropout
|
566 |
+
self.resblock = resblock
|
567 |
+
self.resblock_kernel_sizes = resblock_kernel_sizes
|
568 |
+
self.resblock_dilation_sizes = resblock_dilation_sizes
|
569 |
+
self.upsample_rates = upsample_rates
|
570 |
+
self.upsample_initial_channel = upsample_initial_channel
|
571 |
+
self.upsample_kernel_sizes = upsample_kernel_sizes
|
572 |
+
self.segment_size = segment_size
|
573 |
+
self.gin_channels = gin_channels
|
574 |
+
# self.hop_length = hop_length#
|
575 |
+
self.spk_embed_dim = spk_embed_dim
|
576 |
+
self.enc_p = TextEncoder256(
|
577 |
+
inter_channels,
|
578 |
+
hidden_channels,
|
579 |
+
filter_channels,
|
580 |
+
n_heads,
|
581 |
+
n_layers,
|
582 |
+
kernel_size,
|
583 |
+
p_dropout,
|
584 |
+
)
|
585 |
+
self.dec = GeneratorNSF(
|
586 |
+
inter_channels,
|
587 |
+
resblock,
|
588 |
+
resblock_kernel_sizes,
|
589 |
+
resblock_dilation_sizes,
|
590 |
+
upsample_rates,
|
591 |
+
upsample_initial_channel,
|
592 |
+
upsample_kernel_sizes,
|
593 |
+
gin_channels=gin_channels,
|
594 |
+
sr=sr,
|
595 |
+
is_half=kwargs["is_half"],
|
596 |
+
)
|
597 |
+
self.enc_q = PosteriorEncoder(
|
598 |
+
spec_channels,
|
599 |
+
inter_channels,
|
600 |
+
hidden_channels,
|
601 |
+
5,
|
602 |
+
1,
|
603 |
+
16,
|
604 |
+
gin_channels=gin_channels,
|
605 |
+
)
|
606 |
+
self.flow = ResidualCouplingBlock(
|
607 |
+
inter_channels, hidden_channels, 5, 1, 3, gin_channels=gin_channels
|
608 |
+
)
|
609 |
+
self.emb_g = nn.Embedding(self.spk_embed_dim, gin_channels)
|
610 |
+
print("gin_channels:", gin_channels, "self.spk_embed_dim:", self.spk_embed_dim)
|
611 |
+
|
612 |
+
def remove_weight_norm(self):
|
613 |
+
self.dec.remove_weight_norm()
|
614 |
+
self.flow.remove_weight_norm()
|
615 |
+
self.enc_q.remove_weight_norm()
|
616 |
+
|
617 |
+
def forward(
|
618 |
+
self, phone, phone_lengths, pitch, pitchf, y, y_lengths, ds
|
619 |
+
): # 这里ds是id,[bs,1]
|
620 |
+
# print(1,pitch.shape)#[bs,t]
|
621 |
+
g = self.emb_g(ds).unsqueeze(-1) # [b, 256, 1]##1是t,广播的
|
622 |
+
m_p, logs_p, x_mask = self.enc_p(phone, pitch, phone_lengths)
|
623 |
+
z, m_q, logs_q, y_mask = self.enc_q(y, y_lengths, g=g)
|
624 |
+
z_p = self.flow(z, y_mask, g=g)
|
625 |
+
z_slice, ids_slice = commons.rand_slice_segments(
|
626 |
+
z, y_lengths, self.segment_size
|
627 |
+
)
|
628 |
+
# print(-1,pitchf.shape,ids_slice,self.segment_size,self.hop_length,self.segment_size//self.hop_length)
|
629 |
+
pitchf = commons.slice_segments2(pitchf, ids_slice, self.segment_size)
|
630 |
+
# print(-2,pitchf.shape,z_slice.shape)
|
631 |
+
o = self.dec(z_slice, pitchf, g=g)
|
632 |
+
return o, ids_slice, x_mask, y_mask, (z, z_p, m_p, logs_p, m_q, logs_q)
|
633 |
+
|
634 |
+
def infer(self, phone, phone_lengths, pitch, nsff0, sid, rate=None):
|
635 |
+
g = self.emb_g(sid).unsqueeze(-1)
|
636 |
+
m_p, logs_p, x_mask = self.enc_p(phone, pitch, phone_lengths)
|
637 |
+
z_p = (m_p + torch.exp(logs_p) * torch.randn_like(m_p) * 0.66666) * x_mask
|
638 |
+
if rate:
|
639 |
+
head = int(z_p.shape[2] * rate)
|
640 |
+
z_p = z_p[:, :, -head:]
|
641 |
+
x_mask = x_mask[:, :, -head:]
|
642 |
+
nsff0 = nsff0[:, -head:]
|
643 |
+
z = self.flow(z_p, x_mask, g=g, reverse=True)
|
644 |
+
o = self.dec(z * x_mask, nsff0, g=g)
|
645 |
+
return o, x_mask, (z, z_p, m_p, logs_p)
|
646 |
+
|
647 |
+
|
648 |
+
class SynthesizerTrnMs768NSFsid(nn.Module):
|
649 |
+
def __init__(
|
650 |
+
self,
|
651 |
+
spec_channels,
|
652 |
+
segment_size,
|
653 |
+
inter_channels,
|
654 |
+
hidden_channels,
|
655 |
+
filter_channels,
|
656 |
+
n_heads,
|
657 |
+
n_layers,
|
658 |
+
kernel_size,
|
659 |
+
p_dropout,
|
660 |
+
resblock,
|
661 |
+
resblock_kernel_sizes,
|
662 |
+
resblock_dilation_sizes,
|
663 |
+
upsample_rates,
|
664 |
+
upsample_initial_channel,
|
665 |
+
upsample_kernel_sizes,
|
666 |
+
spk_embed_dim,
|
667 |
+
gin_channels,
|
668 |
+
sr,
|
669 |
+
**kwargs
|
670 |
+
):
|
671 |
+
super().__init__()
|
672 |
+
if type(sr) == type("strr"):
|
673 |
+
sr = sr2sr[sr]
|
674 |
+
self.spec_channels = spec_channels
|
675 |
+
self.inter_channels = inter_channels
|
676 |
+
self.hidden_channels = hidden_channels
|
677 |
+
self.filter_channels = filter_channels
|
678 |
+
self.n_heads = n_heads
|
679 |
+
self.n_layers = n_layers
|
680 |
+
self.kernel_size = kernel_size
|
681 |
+
self.p_dropout = p_dropout
|
682 |
+
self.resblock = resblock
|
683 |
+
self.resblock_kernel_sizes = resblock_kernel_sizes
|
684 |
+
self.resblock_dilation_sizes = resblock_dilation_sizes
|
685 |
+
self.upsample_rates = upsample_rates
|
686 |
+
self.upsample_initial_channel = upsample_initial_channel
|
687 |
+
self.upsample_kernel_sizes = upsample_kernel_sizes
|
688 |
+
self.segment_size = segment_size
|
689 |
+
self.gin_channels = gin_channels
|
690 |
+
# self.hop_length = hop_length#
|
691 |
+
self.spk_embed_dim = spk_embed_dim
|
692 |
+
self.enc_p = TextEncoder768(
|
693 |
+
inter_channels,
|
694 |
+
hidden_channels,
|
695 |
+
filter_channels,
|
696 |
+
n_heads,
|
697 |
+
n_layers,
|
698 |
+
kernel_size,
|
699 |
+
p_dropout,
|
700 |
+
)
|
701 |
+
self.dec = GeneratorNSF(
|
702 |
+
inter_channels,
|
703 |
+
resblock,
|
704 |
+
resblock_kernel_sizes,
|
705 |
+
resblock_dilation_sizes,
|
706 |
+
upsample_rates,
|
707 |
+
upsample_initial_channel,
|
708 |
+
upsample_kernel_sizes,
|
709 |
+
gin_channels=gin_channels,
|
710 |
+
sr=sr,
|
711 |
+
is_half=kwargs["is_half"],
|
712 |
+
)
|
713 |
+
self.enc_q = PosteriorEncoder(
|
714 |
+
spec_channels,
|
715 |
+
inter_channels,
|
716 |
+
hidden_channels,
|
717 |
+
5,
|
718 |
+
1,
|
719 |
+
16,
|
720 |
+
gin_channels=gin_channels,
|
721 |
+
)
|
722 |
+
self.flow = ResidualCouplingBlock(
|
723 |
+
inter_channels, hidden_channels, 5, 1, 3, gin_channels=gin_channels
|
724 |
+
)
|
725 |
+
self.emb_g = nn.Embedding(self.spk_embed_dim, gin_channels)
|
726 |
+
print("gin_channels:", gin_channels, "self.spk_embed_dim:", self.spk_embed_dim)
|
727 |
+
|
728 |
+
def remove_weight_norm(self):
|
729 |
+
self.dec.remove_weight_norm()
|
730 |
+
self.flow.remove_weight_norm()
|
731 |
+
self.enc_q.remove_weight_norm()
|
732 |
+
|
733 |
+
def forward(
|
734 |
+
self, phone, phone_lengths, pitch, pitchf, y, y_lengths, ds
|
735 |
+
): # 这里ds是id,[bs,1]
|
736 |
+
# print(1,pitch.shape)#[bs,t]
|
737 |
+
g = self.emb_g(ds).unsqueeze(-1) # [b, 256, 1]##1是t,广播的
|
738 |
+
m_p, logs_p, x_mask = self.enc_p(phone, pitch, phone_lengths)
|
739 |
+
z, m_q, logs_q, y_mask = self.enc_q(y, y_lengths, g=g)
|
740 |
+
z_p = self.flow(z, y_mask, g=g)
|
741 |
+
z_slice, ids_slice = commons.rand_slice_segments(
|
742 |
+
z, y_lengths, self.segment_size
|
743 |
+
)
|
744 |
+
# print(-1,pitchf.shape,ids_slice,self.segment_size,self.hop_length,self.segment_size//self.hop_length)
|
745 |
+
pitchf = commons.slice_segments2(pitchf, ids_slice, self.segment_size)
|
746 |
+
# print(-2,pitchf.shape,z_slice.shape)
|
747 |
+
o = self.dec(z_slice, pitchf, g=g)
|
748 |
+
return o, ids_slice, x_mask, y_mask, (z, z_p, m_p, logs_p, m_q, logs_q)
|
749 |
+
|
750 |
+
def infer(self, phone, phone_lengths, pitch, nsff0, sid, rate=None):
|
751 |
+
g = self.emb_g(sid).unsqueeze(-1)
|
752 |
+
m_p, logs_p, x_mask = self.enc_p(phone, pitch, phone_lengths)
|
753 |
+
z_p = (m_p + torch.exp(logs_p) * torch.randn_like(m_p) * 0.66666) * x_mask
|
754 |
+
if rate:
|
755 |
+
head = int(z_p.shape[2] * rate)
|
756 |
+
z_p = z_p[:, :, -head:]
|
757 |
+
x_mask = x_mask[:, :, -head:]
|
758 |
+
nsff0 = nsff0[:, -head:]
|
759 |
+
z = self.flow(z_p, x_mask, g=g, reverse=True)
|
760 |
+
o = self.dec(z * x_mask, nsff0, g=g)
|
761 |
+
return o, x_mask, (z, z_p, m_p, logs_p)
|
762 |
+
|
763 |
+
|
764 |
+
class SynthesizerTrnMs256NSFsid_nono(nn.Module):
|
765 |
+
def __init__(
|
766 |
+
self,
|
767 |
+
spec_channels,
|
768 |
+
segment_size,
|
769 |
+
inter_channels,
|
770 |
+
hidden_channels,
|
771 |
+
filter_channels,
|
772 |
+
n_heads,
|
773 |
+
n_layers,
|
774 |
+
kernel_size,
|
775 |
+
p_dropout,
|
776 |
+
resblock,
|
777 |
+
resblock_kernel_sizes,
|
778 |
+
resblock_dilation_sizes,
|
779 |
+
upsample_rates,
|
780 |
+
upsample_initial_channel,
|
781 |
+
upsample_kernel_sizes,
|
782 |
+
spk_embed_dim,
|
783 |
+
gin_channels,
|
784 |
+
sr=None,
|
785 |
+
**kwargs
|
786 |
+
):
|
787 |
+
super().__init__()
|
788 |
+
self.spec_channels = spec_channels
|
789 |
+
self.inter_channels = inter_channels
|
790 |
+
self.hidden_channels = hidden_channels
|
791 |
+
self.filter_channels = filter_channels
|
792 |
+
self.n_heads = n_heads
|
793 |
+
self.n_layers = n_layers
|
794 |
+
self.kernel_size = kernel_size
|
795 |
+
self.p_dropout = p_dropout
|
796 |
+
self.resblock = resblock
|
797 |
+
self.resblock_kernel_sizes = resblock_kernel_sizes
|
798 |
+
self.resblock_dilation_sizes = resblock_dilation_sizes
|
799 |
+
self.upsample_rates = upsample_rates
|
800 |
+
self.upsample_initial_channel = upsample_initial_channel
|
801 |
+
self.upsample_kernel_sizes = upsample_kernel_sizes
|
802 |
+
self.segment_size = segment_size
|
803 |
+
self.gin_channels = gin_channels
|
804 |
+
# self.hop_length = hop_length#
|
805 |
+
self.spk_embed_dim = spk_embed_dim
|
806 |
+
self.enc_p = TextEncoder256(
|
807 |
+
inter_channels,
|
808 |
+
hidden_channels,
|
809 |
+
filter_channels,
|
810 |
+
n_heads,
|
811 |
+
n_layers,
|
812 |
+
kernel_size,
|
813 |
+
p_dropout,
|
814 |
+
f0=False,
|
815 |
+
)
|
816 |
+
self.dec = Generator(
|
817 |
+
inter_channels,
|
818 |
+
resblock,
|
819 |
+
resblock_kernel_sizes,
|
820 |
+
resblock_dilation_sizes,
|
821 |
+
upsample_rates,
|
822 |
+
upsample_initial_channel,
|
823 |
+
upsample_kernel_sizes,
|
824 |
+
gin_channels=gin_channels,
|
825 |
+
)
|
826 |
+
self.enc_q = PosteriorEncoder(
|
827 |
+
spec_channels,
|
828 |
+
inter_channels,
|
829 |
+
hidden_channels,
|
830 |
+
5,
|
831 |
+
1,
|
832 |
+
16,
|
833 |
+
gin_channels=gin_channels,
|
834 |
+
)
|
835 |
+
self.flow = ResidualCouplingBlock(
|
836 |
+
inter_channels, hidden_channels, 5, 1, 3, gin_channels=gin_channels
|
837 |
+
)
|
838 |
+
self.emb_g = nn.Embedding(self.spk_embed_dim, gin_channels)
|
839 |
+
print("gin_channels:", gin_channels, "self.spk_embed_dim:", self.spk_embed_dim)
|
840 |
+
|
841 |
+
def remove_weight_norm(self):
|
842 |
+
self.dec.remove_weight_norm()
|
843 |
+
self.flow.remove_weight_norm()
|
844 |
+
self.enc_q.remove_weight_norm()
|
845 |
+
|
846 |
+
def forward(self, phone, phone_lengths, y, y_lengths, ds): # 这里ds是id,[bs,1]
|
847 |
+
g = self.emb_g(ds).unsqueeze(-1) # [b, 256, 1]##1是t,广播的
|
848 |
+
m_p, logs_p, x_mask = self.enc_p(phone, None, phone_lengths)
|
849 |
+
z, m_q, logs_q, y_mask = self.enc_q(y, y_lengths, g=g)
|
850 |
+
z_p = self.flow(z, y_mask, g=g)
|
851 |
+
z_slice, ids_slice = commons.rand_slice_segments(
|
852 |
+
z, y_lengths, self.segment_size
|
853 |
+
)
|
854 |
+
o = self.dec(z_slice, g=g)
|
855 |
+
return o, ids_slice, x_mask, y_mask, (z, z_p, m_p, logs_p, m_q, logs_q)
|
856 |
+
|
857 |
+
def infer(self, phone, phone_lengths, sid, rate=None):
|
858 |
+
g = self.emb_g(sid).unsqueeze(-1)
|
859 |
+
m_p, logs_p, x_mask = self.enc_p(phone, None, phone_lengths)
|
860 |
+
z_p = (m_p + torch.exp(logs_p) * torch.randn_like(m_p) * 0.66666) * x_mask
|
861 |
+
if rate:
|
862 |
+
head = int(z_p.shape[2] * rate)
|
863 |
+
z_p = z_p[:, :, -head:]
|
864 |
+
x_mask = x_mask[:, :, -head:]
|
865 |
+
z = self.flow(z_p, x_mask, g=g, reverse=True)
|
866 |
+
o = self.dec(z * x_mask, g=g)
|
867 |
+
return o, x_mask, (z, z_p, m_p, logs_p)
|
868 |
+
|
869 |
+
|
870 |
+
class SynthesizerTrnMs768NSFsid_nono(nn.Module):
|
871 |
+
def __init__(
|
872 |
+
self,
|
873 |
+
spec_channels,
|
874 |
+
segment_size,
|
875 |
+
inter_channels,
|
876 |
+
hidden_channels,
|
877 |
+
filter_channels,
|
878 |
+
n_heads,
|
879 |
+
n_layers,
|
880 |
+
kernel_size,
|
881 |
+
p_dropout,
|
882 |
+
resblock,
|
883 |
+
resblock_kernel_sizes,
|
884 |
+
resblock_dilation_sizes,
|
885 |
+
upsample_rates,
|
886 |
+
upsample_initial_channel,
|
887 |
+
upsample_kernel_sizes,
|
888 |
+
spk_embed_dim,
|
889 |
+
gin_channels,
|
890 |
+
sr=None,
|
891 |
+
**kwargs
|
892 |
+
):
|
893 |
+
super().__init__()
|
894 |
+
self.spec_channels = spec_channels
|
895 |
+
self.inter_channels = inter_channels
|
896 |
+
self.hidden_channels = hidden_channels
|
897 |
+
self.filter_channels = filter_channels
|
898 |
+
self.n_heads = n_heads
|
899 |
+
self.n_layers = n_layers
|
900 |
+
self.kernel_size = kernel_size
|
901 |
+
self.p_dropout = p_dropout
|
902 |
+
self.resblock = resblock
|
903 |
+
self.resblock_kernel_sizes = resblock_kernel_sizes
|
904 |
+
self.resblock_dilation_sizes = resblock_dilation_sizes
|
905 |
+
self.upsample_rates = upsample_rates
|
906 |
+
self.upsample_initial_channel = upsample_initial_channel
|
907 |
+
self.upsample_kernel_sizes = upsample_kernel_sizes
|
908 |
+
self.segment_size = segment_size
|
909 |
+
self.gin_channels = gin_channels
|
910 |
+
# self.hop_length = hop_length#
|
911 |
+
self.spk_embed_dim = spk_embed_dim
|
912 |
+
self.enc_p = TextEncoder768(
|
913 |
+
inter_channels,
|
914 |
+
hidden_channels,
|
915 |
+
filter_channels,
|
916 |
+
n_heads,
|
917 |
+
n_layers,
|
918 |
+
kernel_size,
|
919 |
+
p_dropout,
|
920 |
+
f0=False,
|
921 |
+
)
|
922 |
+
self.dec = Generator(
|
923 |
+
inter_channels,
|
924 |
+
resblock,
|
925 |
+
resblock_kernel_sizes,
|
926 |
+
resblock_dilation_sizes,
|
927 |
+
upsample_rates,
|
928 |
+
upsample_initial_channel,
|
929 |
+
upsample_kernel_sizes,
|
930 |
+
gin_channels=gin_channels,
|
931 |
+
)
|
932 |
+
self.enc_q = PosteriorEncoder(
|
933 |
+
spec_channels,
|
934 |
+
inter_channels,
|
935 |
+
hidden_channels,
|
936 |
+
5,
|
937 |
+
1,
|
938 |
+
16,
|
939 |
+
gin_channels=gin_channels,
|
940 |
+
)
|
941 |
+
self.flow = ResidualCouplingBlock(
|
942 |
+
inter_channels, hidden_channels, 5, 1, 3, gin_channels=gin_channels
|
943 |
+
)
|
944 |
+
self.emb_g = nn.Embedding(self.spk_embed_dim, gin_channels)
|
945 |
+
print("gin_channels:", gin_channels, "self.spk_embed_dim:", self.spk_embed_dim)
|
946 |
+
|
947 |
+
def remove_weight_norm(self):
|
948 |
+
self.dec.remove_weight_norm()
|
949 |
+
self.flow.remove_weight_norm()
|
950 |
+
self.enc_q.remove_weight_norm()
|
951 |
+
|
952 |
+
def forward(self, phone, phone_lengths, y, y_lengths, ds): # 这里ds是id,[bs,1]
|
953 |
+
g = self.emb_g(ds).unsqueeze(-1) # [b, 256, 1]##1是t,广播的
|
954 |
+
m_p, logs_p, x_mask = self.enc_p(phone, None, phone_lengths)
|
955 |
+
z, m_q, logs_q, y_mask = self.enc_q(y, y_lengths, g=g)
|
956 |
+
z_p = self.flow(z, y_mask, g=g)
|
957 |
+
z_slice, ids_slice = commons.rand_slice_segments(
|
958 |
+
z, y_lengths, self.segment_size
|
959 |
+
)
|
960 |
+
o = self.dec(z_slice, g=g)
|
961 |
+
return o, ids_slice, x_mask, y_mask, (z, z_p, m_p, logs_p, m_q, logs_q)
|
962 |
+
|
963 |
+
def infer(self, phone, phone_lengths, sid, rate=None):
|
964 |
+
g = self.emb_g(sid).unsqueeze(-1)
|
965 |
+
m_p, logs_p, x_mask = self.enc_p(phone, None, phone_lengths)
|
966 |
+
z_p = (m_p + torch.exp(logs_p) * torch.randn_like(m_p) * 0.66666) * x_mask
|
967 |
+
if rate:
|
968 |
+
head = int(z_p.shape[2] * rate)
|
969 |
+
z_p = z_p[:, :, -head:]
|
970 |
+
x_mask = x_mask[:, :, -head:]
|
971 |
+
z = self.flow(z_p, x_mask, g=g, reverse=True)
|
972 |
+
o = self.dec(z * x_mask, g=g)
|
973 |
+
return o, x_mask, (z, z_p, m_p, logs_p)
|
974 |
+
|
975 |
+
|
976 |
+
class MultiPeriodDiscriminator(torch.nn.Module):
|
977 |
+
def __init__(self, use_spectral_norm=False):
|
978 |
+
super(MultiPeriodDiscriminator, self).__init__()
|
979 |
+
periods = [2, 3, 5, 7, 11, 17]
|
980 |
+
# periods = [3, 5, 7, 11, 17, 23, 37]
|
981 |
+
|
982 |
+
discs = [DiscriminatorS(use_spectral_norm=use_spectral_norm)]
|
983 |
+
discs = discs + [
|
984 |
+
DiscriminatorP(i, use_spectral_norm=use_spectral_norm) for i in periods
|
985 |
+
]
|
986 |
+
self.discriminators = nn.ModuleList(discs)
|
987 |
+
|
988 |
+
def forward(self, y, y_hat):
|
989 |
+
y_d_rs = [] #
|
990 |
+
y_d_gs = []
|
991 |
+
fmap_rs = []
|
992 |
+
fmap_gs = []
|
993 |
+
for i, d in enumerate(self.discriminators):
|
994 |
+
y_d_r, fmap_r = d(y)
|
995 |
+
y_d_g, fmap_g = d(y_hat)
|
996 |
+
# for j in range(len(fmap_r)):
|
997 |
+
# print(i,j,y.shape,y_hat.shape,fmap_r[j].shape,fmap_g[j].shape)
|
998 |
+
y_d_rs.append(y_d_r)
|
999 |
+
y_d_gs.append(y_d_g)
|
1000 |
+
fmap_rs.append(fmap_r)
|
1001 |
+
fmap_gs.append(fmap_g)
|
1002 |
+
|
1003 |
+
return y_d_rs, y_d_gs, fmap_rs, fmap_gs
|
1004 |
+
|
1005 |
+
|
1006 |
+
class MultiPeriodDiscriminatorV2(torch.nn.Module):
|
1007 |
+
def __init__(self, use_spectral_norm=False):
|
1008 |
+
super(MultiPeriodDiscriminatorV2, self).__init__()
|
1009 |
+
# periods = [2, 3, 5, 7, 11, 17]
|
1010 |
+
periods = [2, 3, 5, 7, 11, 17, 23, 37]
|
1011 |
+
|
1012 |
+
discs = [DiscriminatorS(use_spectral_norm=use_spectral_norm)]
|
1013 |
+
discs = discs + [
|
1014 |
+
DiscriminatorP(i, use_spectral_norm=use_spectral_norm) for i in periods
|
1015 |
+
]
|
1016 |
+
self.discriminators = nn.ModuleList(discs)
|
1017 |
+
|
1018 |
+
def forward(self, y, y_hat):
|
1019 |
+
y_d_rs = [] #
|
1020 |
+
y_d_gs = []
|
1021 |
+
fmap_rs = []
|
1022 |
+
fmap_gs = []
|
1023 |
+
for i, d in enumerate(self.discriminators):
|
1024 |
+
y_d_r, fmap_r = d(y)
|
1025 |
+
y_d_g, fmap_g = d(y_hat)
|
1026 |
+
# for j in range(len(fmap_r)):
|
1027 |
+
# print(i,j,y.shape,y_hat.shape,fmap_r[j].shape,fmap_g[j].shape)
|
1028 |
+
y_d_rs.append(y_d_r)
|
1029 |
+
y_d_gs.append(y_d_g)
|
1030 |
+
fmap_rs.append(fmap_r)
|
1031 |
+
fmap_gs.append(fmap_g)
|
1032 |
+
|
1033 |
+
return y_d_rs, y_d_gs, fmap_rs, fmap_gs
|
1034 |
+
|
1035 |
+
|
1036 |
+
class DiscriminatorS(torch.nn.Module):
|
1037 |
+
def __init__(self, use_spectral_norm=False):
|
1038 |
+
super(DiscriminatorS, self).__init__()
|
1039 |
+
norm_f = weight_norm if use_spectral_norm == False else spectral_norm
|
1040 |
+
self.convs = nn.ModuleList(
|
1041 |
+
[
|
1042 |
+
norm_f(Conv1d(1, 16, 15, 1, padding=7)),
|
1043 |
+
norm_f(Conv1d(16, 64, 41, 4, groups=4, padding=20)),
|
1044 |
+
norm_f(Conv1d(64, 256, 41, 4, groups=16, padding=20)),
|
1045 |
+
norm_f(Conv1d(256, 1024, 41, 4, groups=64, padding=20)),
|
1046 |
+
norm_f(Conv1d(1024, 1024, 41, 4, groups=256, padding=20)),
|
1047 |
+
norm_f(Conv1d(1024, 1024, 5, 1, padding=2)),
|
1048 |
+
]
|
1049 |
+
)
|
1050 |
+
self.conv_post = norm_f(Conv1d(1024, 1, 3, 1, padding=1))
|
1051 |
+
|
1052 |
+
def forward(self, x):
|
1053 |
+
fmap = []
|
1054 |
+
|
1055 |
+
for l in self.convs:
|
1056 |
+
x = l(x)
|
1057 |
+
x = F.leaky_relu(x, modules.LRELU_SLOPE)
|
1058 |
+
fmap.append(x)
|
1059 |
+
x = self.conv_post(x)
|
1060 |
+
fmap.append(x)
|
1061 |
+
x = torch.flatten(x, 1, -1)
|
1062 |
+
|
1063 |
+
return x, fmap
|
1064 |
+
|
1065 |
+
|
1066 |
+
class DiscriminatorP(torch.nn.Module):
|
1067 |
+
def __init__(self, period, kernel_size=5, stride=3, use_spectral_norm=False):
|
1068 |
+
super(DiscriminatorP, self).__init__()
|
1069 |
+
self.period = period
|
1070 |
+
self.use_spectral_norm = use_spectral_norm
|
1071 |
+
norm_f = weight_norm if use_spectral_norm == False else spectral_norm
|
1072 |
+
self.convs = nn.ModuleList(
|
1073 |
+
[
|
1074 |
+
norm_f(
|
1075 |
+
Conv2d(
|
1076 |
+
1,
|
1077 |
+
32,
|
1078 |
+
(kernel_size, 1),
|
1079 |
+
(stride, 1),
|
1080 |
+
padding=(get_padding(kernel_size, 1), 0),
|
1081 |
+
)
|
1082 |
+
),
|
1083 |
+
norm_f(
|
1084 |
+
Conv2d(
|
1085 |
+
32,
|
1086 |
+
128,
|
1087 |
+
(kernel_size, 1),
|
1088 |
+
(stride, 1),
|
1089 |
+
padding=(get_padding(kernel_size, 1), 0),
|
1090 |
+
)
|
1091 |
+
),
|
1092 |
+
norm_f(
|
1093 |
+
Conv2d(
|
1094 |
+
128,
|
1095 |
+
512,
|
1096 |
+
(kernel_size, 1),
|
1097 |
+
(stride, 1),
|
1098 |
+
padding=(get_padding(kernel_size, 1), 0),
|
1099 |
+
)
|
1100 |
+
),
|
1101 |
+
norm_f(
|
1102 |
+
Conv2d(
|
1103 |
+
512,
|
1104 |
+
1024,
|
1105 |
+
(kernel_size, 1),
|
1106 |
+
(stride, 1),
|
1107 |
+
padding=(get_padding(kernel_size, 1), 0),
|
1108 |
+
)
|
1109 |
+
),
|
1110 |
+
norm_f(
|
1111 |
+
Conv2d(
|
1112 |
+
1024,
|
1113 |
+
1024,
|
1114 |
+
(kernel_size, 1),
|
1115 |
+
1,
|
1116 |
+
padding=(get_padding(kernel_size, 1), 0),
|
1117 |
+
)
|
1118 |
+
),
|
1119 |
+
]
|
1120 |
+
)
|
1121 |
+
self.conv_post = norm_f(Conv2d(1024, 1, (3, 1), 1, padding=(1, 0)))
|
1122 |
+
|
1123 |
+
def forward(self, x):
|
1124 |
+
fmap = []
|
1125 |
+
|
1126 |
+
# 1d to 2d
|
1127 |
+
b, c, t = x.shape
|
1128 |
+
if t % self.period != 0: # pad first
|
1129 |
+
n_pad = self.period - (t % self.period)
|
1130 |
+
x = F.pad(x, (0, n_pad), "reflect")
|
1131 |
+
t = t + n_pad
|
1132 |
+
x = x.view(b, c, t // self.period, self.period)
|
1133 |
+
|
1134 |
+
for l in self.convs:
|
1135 |
+
x = l(x)
|
1136 |
+
x = F.leaky_relu(x, modules.LRELU_SLOPE)
|
1137 |
+
fmap.append(x)
|
1138 |
+
x = self.conv_post(x)
|
1139 |
+
fmap.append(x)
|
1140 |
+
x = torch.flatten(x, 1, -1)
|
1141 |
+
|
1142 |
+
return x, fmap
|
lib/infer_pack/models_onnx.py
ADDED
@@ -0,0 +1,819 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import math, pdb, os
|
2 |
+
from time import time as ttime
|
3 |
+
import torch
|
4 |
+
from torch import nn
|
5 |
+
from torch.nn import functional as F
|
6 |
+
from lib.infer_pack import modules
|
7 |
+
from lib.infer_pack import attentions
|
8 |
+
from lib.infer_pack import commons
|
9 |
+
from lib.infer_pack.commons import init_weights, get_padding
|
10 |
+
from torch.nn import Conv1d, ConvTranspose1d, AvgPool1d, Conv2d
|
11 |
+
from torch.nn.utils import weight_norm, remove_weight_norm, spectral_norm
|
12 |
+
from lib.infer_pack.commons import init_weights
|
13 |
+
import numpy as np
|
14 |
+
from lib.infer_pack import commons
|
15 |
+
|
16 |
+
|
17 |
+
class TextEncoder256(nn.Module):
|
18 |
+
def __init__(
|
19 |
+
self,
|
20 |
+
out_channels,
|
21 |
+
hidden_channels,
|
22 |
+
filter_channels,
|
23 |
+
n_heads,
|
24 |
+
n_layers,
|
25 |
+
kernel_size,
|
26 |
+
p_dropout,
|
27 |
+
f0=True,
|
28 |
+
):
|
29 |
+
super().__init__()
|
30 |
+
self.out_channels = out_channels
|
31 |
+
self.hidden_channels = hidden_channels
|
32 |
+
self.filter_channels = filter_channels
|
33 |
+
self.n_heads = n_heads
|
34 |
+
self.n_layers = n_layers
|
35 |
+
self.kernel_size = kernel_size
|
36 |
+
self.p_dropout = p_dropout
|
37 |
+
self.emb_phone = nn.Linear(256, hidden_channels)
|
38 |
+
self.lrelu = nn.LeakyReLU(0.1, inplace=True)
|
39 |
+
if f0 == True:
|
40 |
+
self.emb_pitch = nn.Embedding(256, hidden_channels) # pitch 256
|
41 |
+
self.encoder = attentions.Encoder(
|
42 |
+
hidden_channels, filter_channels, n_heads, n_layers, kernel_size, p_dropout
|
43 |
+
)
|
44 |
+
self.proj = nn.Conv1d(hidden_channels, out_channels * 2, 1)
|
45 |
+
|
46 |
+
def forward(self, phone, pitch, lengths):
|
47 |
+
if pitch == None:
|
48 |
+
x = self.emb_phone(phone)
|
49 |
+
else:
|
50 |
+
x = self.emb_phone(phone) + self.emb_pitch(pitch)
|
51 |
+
x = x * math.sqrt(self.hidden_channels) # [b, t, h]
|
52 |
+
x = self.lrelu(x)
|
53 |
+
x = torch.transpose(x, 1, -1) # [b, h, t]
|
54 |
+
x_mask = torch.unsqueeze(commons.sequence_mask(lengths, x.size(2)), 1).to(
|
55 |
+
x.dtype
|
56 |
+
)
|
57 |
+
x = self.encoder(x * x_mask, x_mask)
|
58 |
+
stats = self.proj(x) * x_mask
|
59 |
+
|
60 |
+
m, logs = torch.split(stats, self.out_channels, dim=1)
|
61 |
+
return m, logs, x_mask
|
62 |
+
|
63 |
+
|
64 |
+
class TextEncoder768(nn.Module):
|
65 |
+
def __init__(
|
66 |
+
self,
|
67 |
+
out_channels,
|
68 |
+
hidden_channels,
|
69 |
+
filter_channels,
|
70 |
+
n_heads,
|
71 |
+
n_layers,
|
72 |
+
kernel_size,
|
73 |
+
p_dropout,
|
74 |
+
f0=True,
|
75 |
+
):
|
76 |
+
super().__init__()
|
77 |
+
self.out_channels = out_channels
|
78 |
+
self.hidden_channels = hidden_channels
|
79 |
+
self.filter_channels = filter_channels
|
80 |
+
self.n_heads = n_heads
|
81 |
+
self.n_layers = n_layers
|
82 |
+
self.kernel_size = kernel_size
|
83 |
+
self.p_dropout = p_dropout
|
84 |
+
self.emb_phone = nn.Linear(768, hidden_channels)
|
85 |
+
self.lrelu = nn.LeakyReLU(0.1, inplace=True)
|
86 |
+
if f0 == True:
|
87 |
+
self.emb_pitch = nn.Embedding(256, hidden_channels) # pitch 256
|
88 |
+
self.encoder = attentions.Encoder(
|
89 |
+
hidden_channels, filter_channels, n_heads, n_layers, kernel_size, p_dropout
|
90 |
+
)
|
91 |
+
self.proj = nn.Conv1d(hidden_channels, out_channels * 2, 1)
|
92 |
+
|
93 |
+
def forward(self, phone, pitch, lengths):
|
94 |
+
if pitch == None:
|
95 |
+
x = self.emb_phone(phone)
|
96 |
+
else:
|
97 |
+
x = self.emb_phone(phone) + self.emb_pitch(pitch)
|
98 |
+
x = x * math.sqrt(self.hidden_channels) # [b, t, h]
|
99 |
+
x = self.lrelu(x)
|
100 |
+
x = torch.transpose(x, 1, -1) # [b, h, t]
|
101 |
+
x_mask = torch.unsqueeze(commons.sequence_mask(lengths, x.size(2)), 1).to(
|
102 |
+
x.dtype
|
103 |
+
)
|
104 |
+
x = self.encoder(x * x_mask, x_mask)
|
105 |
+
stats = self.proj(x) * x_mask
|
106 |
+
|
107 |
+
m, logs = torch.split(stats, self.out_channels, dim=1)
|
108 |
+
return m, logs, x_mask
|
109 |
+
|
110 |
+
|
111 |
+
class ResidualCouplingBlock(nn.Module):
|
112 |
+
def __init__(
|
113 |
+
self,
|
114 |
+
channels,
|
115 |
+
hidden_channels,
|
116 |
+
kernel_size,
|
117 |
+
dilation_rate,
|
118 |
+
n_layers,
|
119 |
+
n_flows=4,
|
120 |
+
gin_channels=0,
|
121 |
+
):
|
122 |
+
super().__init__()
|
123 |
+
self.channels = channels
|
124 |
+
self.hidden_channels = hidden_channels
|
125 |
+
self.kernel_size = kernel_size
|
126 |
+
self.dilation_rate = dilation_rate
|
127 |
+
self.n_layers = n_layers
|
128 |
+
self.n_flows = n_flows
|
129 |
+
self.gin_channels = gin_channels
|
130 |
+
|
131 |
+
self.flows = nn.ModuleList()
|
132 |
+
for i in range(n_flows):
|
133 |
+
self.flows.append(
|
134 |
+
modules.ResidualCouplingLayer(
|
135 |
+
channels,
|
136 |
+
hidden_channels,
|
137 |
+
kernel_size,
|
138 |
+
dilation_rate,
|
139 |
+
n_layers,
|
140 |
+
gin_channels=gin_channels,
|
141 |
+
mean_only=True,
|
142 |
+
)
|
143 |
+
)
|
144 |
+
self.flows.append(modules.Flip())
|
145 |
+
|
146 |
+
def forward(self, x, x_mask, g=None, reverse=False):
|
147 |
+
if not reverse:
|
148 |
+
for flow in self.flows:
|
149 |
+
x, _ = flow(x, x_mask, g=g, reverse=reverse)
|
150 |
+
else:
|
151 |
+
for flow in reversed(self.flows):
|
152 |
+
x = flow(x, x_mask, g=g, reverse=reverse)
|
153 |
+
return x
|
154 |
+
|
155 |
+
def remove_weight_norm(self):
|
156 |
+
for i in range(self.n_flows):
|
157 |
+
self.flows[i * 2].remove_weight_norm()
|
158 |
+
|
159 |
+
|
160 |
+
class PosteriorEncoder(nn.Module):
|
161 |
+
def __init__(
|
162 |
+
self,
|
163 |
+
in_channels,
|
164 |
+
out_channels,
|
165 |
+
hidden_channels,
|
166 |
+
kernel_size,
|
167 |
+
dilation_rate,
|
168 |
+
n_layers,
|
169 |
+
gin_channels=0,
|
170 |
+
):
|
171 |
+
super().__init__()
|
172 |
+
self.in_channels = in_channels
|
173 |
+
self.out_channels = out_channels
|
174 |
+
self.hidden_channels = hidden_channels
|
175 |
+
self.kernel_size = kernel_size
|
176 |
+
self.dilation_rate = dilation_rate
|
177 |
+
self.n_layers = n_layers
|
178 |
+
self.gin_channels = gin_channels
|
179 |
+
|
180 |
+
self.pre = nn.Conv1d(in_channels, hidden_channels, 1)
|
181 |
+
self.enc = modules.WN(
|
182 |
+
hidden_channels,
|
183 |
+
kernel_size,
|
184 |
+
dilation_rate,
|
185 |
+
n_layers,
|
186 |
+
gin_channels=gin_channels,
|
187 |
+
)
|
188 |
+
self.proj = nn.Conv1d(hidden_channels, out_channels * 2, 1)
|
189 |
+
|
190 |
+
def forward(self, x, x_lengths, g=None):
|
191 |
+
x_mask = torch.unsqueeze(commons.sequence_mask(x_lengths, x.size(2)), 1).to(
|
192 |
+
x.dtype
|
193 |
+
)
|
194 |
+
x = self.pre(x) * x_mask
|
195 |
+
x = self.enc(x, x_mask, g=g)
|
196 |
+
stats = self.proj(x) * x_mask
|
197 |
+
m, logs = torch.split(stats, self.out_channels, dim=1)
|
198 |
+
z = (m + torch.randn_like(m) * torch.exp(logs)) * x_mask
|
199 |
+
return z, m, logs, x_mask
|
200 |
+
|
201 |
+
def remove_weight_norm(self):
|
202 |
+
self.enc.remove_weight_norm()
|
203 |
+
|
204 |
+
|
205 |
+
class Generator(torch.nn.Module):
|
206 |
+
def __init__(
|
207 |
+
self,
|
208 |
+
initial_channel,
|
209 |
+
resblock,
|
210 |
+
resblock_kernel_sizes,
|
211 |
+
resblock_dilation_sizes,
|
212 |
+
upsample_rates,
|
213 |
+
upsample_initial_channel,
|
214 |
+
upsample_kernel_sizes,
|
215 |
+
gin_channels=0,
|
216 |
+
):
|
217 |
+
super(Generator, self).__init__()
|
218 |
+
self.num_kernels = len(resblock_kernel_sizes)
|
219 |
+
self.num_upsamples = len(upsample_rates)
|
220 |
+
self.conv_pre = Conv1d(
|
221 |
+
initial_channel, upsample_initial_channel, 7, 1, padding=3
|
222 |
+
)
|
223 |
+
resblock = modules.ResBlock1 if resblock == "1" else modules.ResBlock2
|
224 |
+
|
225 |
+
self.ups = nn.ModuleList()
|
226 |
+
for i, (u, k) in enumerate(zip(upsample_rates, upsample_kernel_sizes)):
|
227 |
+
self.ups.append(
|
228 |
+
weight_norm(
|
229 |
+
ConvTranspose1d(
|
230 |
+
upsample_initial_channel // (2**i),
|
231 |
+
upsample_initial_channel // (2 ** (i + 1)),
|
232 |
+
k,
|
233 |
+
u,
|
234 |
+
padding=(k - u) // 2,
|
235 |
+
)
|
236 |
+
)
|
237 |
+
)
|
238 |
+
|
239 |
+
self.resblocks = nn.ModuleList()
|
240 |
+
for i in range(len(self.ups)):
|
241 |
+
ch = upsample_initial_channel // (2 ** (i + 1))
|
242 |
+
for j, (k, d) in enumerate(
|
243 |
+
zip(resblock_kernel_sizes, resblock_dilation_sizes)
|
244 |
+
):
|
245 |
+
self.resblocks.append(resblock(ch, k, d))
|
246 |
+
|
247 |
+
self.conv_post = Conv1d(ch, 1, 7, 1, padding=3, bias=False)
|
248 |
+
self.ups.apply(init_weights)
|
249 |
+
|
250 |
+
if gin_channels != 0:
|
251 |
+
self.cond = nn.Conv1d(gin_channels, upsample_initial_channel, 1)
|
252 |
+
|
253 |
+
def forward(self, x, g=None):
|
254 |
+
x = self.conv_pre(x)
|
255 |
+
if g is not None:
|
256 |
+
x = x + self.cond(g)
|
257 |
+
|
258 |
+
for i in range(self.num_upsamples):
|
259 |
+
x = F.leaky_relu(x, modules.LRELU_SLOPE)
|
260 |
+
x = self.ups[i](x)
|
261 |
+
xs = None
|
262 |
+
for j in range(self.num_kernels):
|
263 |
+
if xs is None:
|
264 |
+
xs = self.resblocks[i * self.num_kernels + j](x)
|
265 |
+
else:
|
266 |
+
xs += self.resblocks[i * self.num_kernels + j](x)
|
267 |
+
x = xs / self.num_kernels
|
268 |
+
x = F.leaky_relu(x)
|
269 |
+
x = self.conv_post(x)
|
270 |
+
x = torch.tanh(x)
|
271 |
+
|
272 |
+
return x
|
273 |
+
|
274 |
+
def remove_weight_norm(self):
|
275 |
+
for l in self.ups:
|
276 |
+
remove_weight_norm(l)
|
277 |
+
for l in self.resblocks:
|
278 |
+
l.remove_weight_norm()
|
279 |
+
|
280 |
+
|
281 |
+
class SineGen(torch.nn.Module):
|
282 |
+
"""Definition of sine generator
|
283 |
+
SineGen(samp_rate, harmonic_num = 0,
|
284 |
+
sine_amp = 0.1, noise_std = 0.003,
|
285 |
+
voiced_threshold = 0,
|
286 |
+
flag_for_pulse=False)
|
287 |
+
samp_rate: sampling rate in Hz
|
288 |
+
harmonic_num: number of harmonic overtones (default 0)
|
289 |
+
sine_amp: amplitude of sine-wavefrom (default 0.1)
|
290 |
+
noise_std: std of Gaussian noise (default 0.003)
|
291 |
+
voiced_thoreshold: F0 threshold for U/V classification (default 0)
|
292 |
+
flag_for_pulse: this SinGen is used inside PulseGen (default False)
|
293 |
+
Note: when flag_for_pulse is True, the first time step of a voiced
|
294 |
+
segment is always sin(np.pi) or cos(0)
|
295 |
+
"""
|
296 |
+
|
297 |
+
def __init__(
|
298 |
+
self,
|
299 |
+
samp_rate,
|
300 |
+
harmonic_num=0,
|
301 |
+
sine_amp=0.1,
|
302 |
+
noise_std=0.003,
|
303 |
+
voiced_threshold=0,
|
304 |
+
flag_for_pulse=False,
|
305 |
+
):
|
306 |
+
super(SineGen, self).__init__()
|
307 |
+
self.sine_amp = sine_amp
|
308 |
+
self.noise_std = noise_std
|
309 |
+
self.harmonic_num = harmonic_num
|
310 |
+
self.dim = self.harmonic_num + 1
|
311 |
+
self.sampling_rate = samp_rate
|
312 |
+
self.voiced_threshold = voiced_threshold
|
313 |
+
|
314 |
+
def _f02uv(self, f0):
|
315 |
+
# generate uv signal
|
316 |
+
uv = torch.ones_like(f0)
|
317 |
+
uv = uv * (f0 > self.voiced_threshold)
|
318 |
+
return uv
|
319 |
+
|
320 |
+
def forward(self, f0, upp):
|
321 |
+
"""sine_tensor, uv = forward(f0)
|
322 |
+
input F0: tensor(batchsize=1, length, dim=1)
|
323 |
+
f0 for unvoiced steps should be 0
|
324 |
+
output sine_tensor: tensor(batchsize=1, length, dim)
|
325 |
+
output uv: tensor(batchsize=1, length, 1)
|
326 |
+
"""
|
327 |
+
with torch.no_grad():
|
328 |
+
f0 = f0[:, None].transpose(1, 2)
|
329 |
+
f0_buf = torch.zeros(f0.shape[0], f0.shape[1], self.dim, device=f0.device)
|
330 |
+
# fundamental component
|
331 |
+
f0_buf[:, :, 0] = f0[:, :, 0]
|
332 |
+
for idx in np.arange(self.harmonic_num):
|
333 |
+
f0_buf[:, :, idx + 1] = f0_buf[:, :, 0] * (
|
334 |
+
idx + 2
|
335 |
+
) # idx + 2: the (idx+1)-th overtone, (idx+2)-th harmonic
|
336 |
+
rad_values = (f0_buf / self.sampling_rate) % 1 ###%1意味着n_har的乘积无法后处理优化
|
337 |
+
rand_ini = torch.rand(
|
338 |
+
f0_buf.shape[0], f0_buf.shape[2], device=f0_buf.device
|
339 |
+
)
|
340 |
+
rand_ini[:, 0] = 0
|
341 |
+
rad_values[:, 0, :] = rad_values[:, 0, :] + rand_ini
|
342 |
+
tmp_over_one = torch.cumsum(rad_values, 1) # % 1 #####%1意味着后面的cumsum无法再优化
|
343 |
+
tmp_over_one *= upp
|
344 |
+
tmp_over_one = F.interpolate(
|
345 |
+
tmp_over_one.transpose(2, 1),
|
346 |
+
scale_factor=upp,
|
347 |
+
mode="linear",
|
348 |
+
align_corners=True,
|
349 |
+
).transpose(2, 1)
|
350 |
+
rad_values = F.interpolate(
|
351 |
+
rad_values.transpose(2, 1), scale_factor=upp, mode="nearest"
|
352 |
+
).transpose(
|
353 |
+
2, 1
|
354 |
+
) #######
|
355 |
+
tmp_over_one %= 1
|
356 |
+
tmp_over_one_idx = (tmp_over_one[:, 1:, :] - tmp_over_one[:, :-1, :]) < 0
|
357 |
+
cumsum_shift = torch.zeros_like(rad_values)
|
358 |
+
cumsum_shift[:, 1:, :] = tmp_over_one_idx * -1.0
|
359 |
+
sine_waves = torch.sin(
|
360 |
+
torch.cumsum(rad_values + cumsum_shift, dim=1) * 2 * np.pi
|
361 |
+
)
|
362 |
+
sine_waves = sine_waves * self.sine_amp
|
363 |
+
uv = self._f02uv(f0)
|
364 |
+
uv = F.interpolate(
|
365 |
+
uv.transpose(2, 1), scale_factor=upp, mode="nearest"
|
366 |
+
).transpose(2, 1)
|
367 |
+
noise_amp = uv * self.noise_std + (1 - uv) * self.sine_amp / 3
|
368 |
+
noise = noise_amp * torch.randn_like(sine_waves)
|
369 |
+
sine_waves = sine_waves * uv + noise
|
370 |
+
return sine_waves, uv, noise
|
371 |
+
|
372 |
+
|
373 |
+
class SourceModuleHnNSF(torch.nn.Module):
|
374 |
+
"""SourceModule for hn-nsf
|
375 |
+
SourceModule(sampling_rate, harmonic_num=0, sine_amp=0.1,
|
376 |
+
add_noise_std=0.003, voiced_threshod=0)
|
377 |
+
sampling_rate: sampling_rate in Hz
|
378 |
+
harmonic_num: number of harmonic above F0 (default: 0)
|
379 |
+
sine_amp: amplitude of sine source signal (default: 0.1)
|
380 |
+
add_noise_std: std of additive Gaussian noise (default: 0.003)
|
381 |
+
note that amplitude of noise in unvoiced is decided
|
382 |
+
by sine_amp
|
383 |
+
voiced_threshold: threhold to set U/V given F0 (default: 0)
|
384 |
+
Sine_source, noise_source = SourceModuleHnNSF(F0_sampled)
|
385 |
+
F0_sampled (batchsize, length, 1)
|
386 |
+
Sine_source (batchsize, length, 1)
|
387 |
+
noise_source (batchsize, length 1)
|
388 |
+
uv (batchsize, length, 1)
|
389 |
+
"""
|
390 |
+
|
391 |
+
def __init__(
|
392 |
+
self,
|
393 |
+
sampling_rate,
|
394 |
+
harmonic_num=0,
|
395 |
+
sine_amp=0.1,
|
396 |
+
add_noise_std=0.003,
|
397 |
+
voiced_threshod=0,
|
398 |
+
is_half=True,
|
399 |
+
):
|
400 |
+
super(SourceModuleHnNSF, self).__init__()
|
401 |
+
|
402 |
+
self.sine_amp = sine_amp
|
403 |
+
self.noise_std = add_noise_std
|
404 |
+
self.is_half = is_half
|
405 |
+
# to produce sine waveforms
|
406 |
+
self.l_sin_gen = SineGen(
|
407 |
+
sampling_rate, harmonic_num, sine_amp, add_noise_std, voiced_threshod
|
408 |
+
)
|
409 |
+
|
410 |
+
# to merge source harmonics into a single excitation
|
411 |
+
self.l_linear = torch.nn.Linear(harmonic_num + 1, 1)
|
412 |
+
self.l_tanh = torch.nn.Tanh()
|
413 |
+
|
414 |
+
def forward(self, x, upp=None):
|
415 |
+
sine_wavs, uv, _ = self.l_sin_gen(x, upp)
|
416 |
+
if self.is_half:
|
417 |
+
sine_wavs = sine_wavs.half()
|
418 |
+
sine_merge = self.l_tanh(self.l_linear(sine_wavs))
|
419 |
+
return sine_merge, None, None # noise, uv
|
420 |
+
|
421 |
+
|
422 |
+
class GeneratorNSF(torch.nn.Module):
|
423 |
+
def __init__(
|
424 |
+
self,
|
425 |
+
initial_channel,
|
426 |
+
resblock,
|
427 |
+
resblock_kernel_sizes,
|
428 |
+
resblock_dilation_sizes,
|
429 |
+
upsample_rates,
|
430 |
+
upsample_initial_channel,
|
431 |
+
upsample_kernel_sizes,
|
432 |
+
gin_channels,
|
433 |
+
sr,
|
434 |
+
is_half=False,
|
435 |
+
):
|
436 |
+
super(GeneratorNSF, self).__init__()
|
437 |
+
self.num_kernels = len(resblock_kernel_sizes)
|
438 |
+
self.num_upsamples = len(upsample_rates)
|
439 |
+
|
440 |
+
self.f0_upsamp = torch.nn.Upsample(scale_factor=np.prod(upsample_rates))
|
441 |
+
self.m_source = SourceModuleHnNSF(
|
442 |
+
sampling_rate=sr, harmonic_num=0, is_half=is_half
|
443 |
+
)
|
444 |
+
self.noise_convs = nn.ModuleList()
|
445 |
+
self.conv_pre = Conv1d(
|
446 |
+
initial_channel, upsample_initial_channel, 7, 1, padding=3
|
447 |
+
)
|
448 |
+
resblock = modules.ResBlock1 if resblock == "1" else modules.ResBlock2
|
449 |
+
|
450 |
+
self.ups = nn.ModuleList()
|
451 |
+
for i, (u, k) in enumerate(zip(upsample_rates, upsample_kernel_sizes)):
|
452 |
+
c_cur = upsample_initial_channel // (2 ** (i + 1))
|
453 |
+
self.ups.append(
|
454 |
+
weight_norm(
|
455 |
+
ConvTranspose1d(
|
456 |
+
upsample_initial_channel // (2**i),
|
457 |
+
upsample_initial_channel // (2 ** (i + 1)),
|
458 |
+
k,
|
459 |
+
u,
|
460 |
+
padding=(k - u) // 2,
|
461 |
+
)
|
462 |
+
)
|
463 |
+
)
|
464 |
+
if i + 1 < len(upsample_rates):
|
465 |
+
stride_f0 = np.prod(upsample_rates[i + 1 :])
|
466 |
+
self.noise_convs.append(
|
467 |
+
Conv1d(
|
468 |
+
1,
|
469 |
+
c_cur,
|
470 |
+
kernel_size=stride_f0 * 2,
|
471 |
+
stride=stride_f0,
|
472 |
+
padding=stride_f0 // 2,
|
473 |
+
)
|
474 |
+
)
|
475 |
+
else:
|
476 |
+
self.noise_convs.append(Conv1d(1, c_cur, kernel_size=1))
|
477 |
+
|
478 |
+
self.resblocks = nn.ModuleList()
|
479 |
+
for i in range(len(self.ups)):
|
480 |
+
ch = upsample_initial_channel // (2 ** (i + 1))
|
481 |
+
for j, (k, d) in enumerate(
|
482 |
+
zip(resblock_kernel_sizes, resblock_dilation_sizes)
|
483 |
+
):
|
484 |
+
self.resblocks.append(resblock(ch, k, d))
|
485 |
+
|
486 |
+
self.conv_post = Conv1d(ch, 1, 7, 1, padding=3, bias=False)
|
487 |
+
self.ups.apply(init_weights)
|
488 |
+
|
489 |
+
if gin_channels != 0:
|
490 |
+
self.cond = nn.Conv1d(gin_channels, upsample_initial_channel, 1)
|
491 |
+
|
492 |
+
self.upp = np.prod(upsample_rates)
|
493 |
+
|
494 |
+
def forward(self, x, f0, g=None):
|
495 |
+
har_source, noi_source, uv = self.m_source(f0, self.upp)
|
496 |
+
har_source = har_source.transpose(1, 2)
|
497 |
+
x = self.conv_pre(x)
|
498 |
+
if g is not None:
|
499 |
+
x = x + self.cond(g)
|
500 |
+
|
501 |
+
for i in range(self.num_upsamples):
|
502 |
+
x = F.leaky_relu(x, modules.LRELU_SLOPE)
|
503 |
+
x = self.ups[i](x)
|
504 |
+
x_source = self.noise_convs[i](har_source)
|
505 |
+
x = x + x_source
|
506 |
+
xs = None
|
507 |
+
for j in range(self.num_kernels):
|
508 |
+
if xs is None:
|
509 |
+
xs = self.resblocks[i * self.num_kernels + j](x)
|
510 |
+
else:
|
511 |
+
xs += self.resblocks[i * self.num_kernels + j](x)
|
512 |
+
x = xs / self.num_kernels
|
513 |
+
x = F.leaky_relu(x)
|
514 |
+
x = self.conv_post(x)
|
515 |
+
x = torch.tanh(x)
|
516 |
+
return x
|
517 |
+
|
518 |
+
def remove_weight_norm(self):
|
519 |
+
for l in self.ups:
|
520 |
+
remove_weight_norm(l)
|
521 |
+
for l in self.resblocks:
|
522 |
+
l.remove_weight_norm()
|
523 |
+
|
524 |
+
|
525 |
+
sr2sr = {
|
526 |
+
"32k": 32000,
|
527 |
+
"40k": 40000,
|
528 |
+
"48k": 48000,
|
529 |
+
}
|
530 |
+
|
531 |
+
|
532 |
+
class SynthesizerTrnMsNSFsidM(nn.Module):
|
533 |
+
def __init__(
|
534 |
+
self,
|
535 |
+
spec_channels,
|
536 |
+
segment_size,
|
537 |
+
inter_channels,
|
538 |
+
hidden_channels,
|
539 |
+
filter_channels,
|
540 |
+
n_heads,
|
541 |
+
n_layers,
|
542 |
+
kernel_size,
|
543 |
+
p_dropout,
|
544 |
+
resblock,
|
545 |
+
resblock_kernel_sizes,
|
546 |
+
resblock_dilation_sizes,
|
547 |
+
upsample_rates,
|
548 |
+
upsample_initial_channel,
|
549 |
+
upsample_kernel_sizes,
|
550 |
+
spk_embed_dim,
|
551 |
+
gin_channels,
|
552 |
+
sr,
|
553 |
+
version,
|
554 |
+
**kwargs
|
555 |
+
):
|
556 |
+
super().__init__()
|
557 |
+
if type(sr) == type("strr"):
|
558 |
+
sr = sr2sr[sr]
|
559 |
+
self.spec_channels = spec_channels
|
560 |
+
self.inter_channels = inter_channels
|
561 |
+
self.hidden_channels = hidden_channels
|
562 |
+
self.filter_channels = filter_channels
|
563 |
+
self.n_heads = n_heads
|
564 |
+
self.n_layers = n_layers
|
565 |
+
self.kernel_size = kernel_size
|
566 |
+
self.p_dropout = p_dropout
|
567 |
+
self.resblock = resblock
|
568 |
+
self.resblock_kernel_sizes = resblock_kernel_sizes
|
569 |
+
self.resblock_dilation_sizes = resblock_dilation_sizes
|
570 |
+
self.upsample_rates = upsample_rates
|
571 |
+
self.upsample_initial_channel = upsample_initial_channel
|
572 |
+
self.upsample_kernel_sizes = upsample_kernel_sizes
|
573 |
+
self.segment_size = segment_size
|
574 |
+
self.gin_channels = gin_channels
|
575 |
+
# self.hop_length = hop_length#
|
576 |
+
self.spk_embed_dim = spk_embed_dim
|
577 |
+
if version == "v1":
|
578 |
+
self.enc_p = TextEncoder256(
|
579 |
+
inter_channels,
|
580 |
+
hidden_channels,
|
581 |
+
filter_channels,
|
582 |
+
n_heads,
|
583 |
+
n_layers,
|
584 |
+
kernel_size,
|
585 |
+
p_dropout,
|
586 |
+
)
|
587 |
+
else:
|
588 |
+
self.enc_p = TextEncoder768(
|
589 |
+
inter_channels,
|
590 |
+
hidden_channels,
|
591 |
+
filter_channels,
|
592 |
+
n_heads,
|
593 |
+
n_layers,
|
594 |
+
kernel_size,
|
595 |
+
p_dropout,
|
596 |
+
)
|
597 |
+
self.dec = GeneratorNSF(
|
598 |
+
inter_channels,
|
599 |
+
resblock,
|
600 |
+
resblock_kernel_sizes,
|
601 |
+
resblock_dilation_sizes,
|
602 |
+
upsample_rates,
|
603 |
+
upsample_initial_channel,
|
604 |
+
upsample_kernel_sizes,
|
605 |
+
gin_channels=gin_channels,
|
606 |
+
sr=sr,
|
607 |
+
is_half=kwargs["is_half"],
|
608 |
+
)
|
609 |
+
self.enc_q = PosteriorEncoder(
|
610 |
+
spec_channels,
|
611 |
+
inter_channels,
|
612 |
+
hidden_channels,
|
613 |
+
5,
|
614 |
+
1,
|
615 |
+
16,
|
616 |
+
gin_channels=gin_channels,
|
617 |
+
)
|
618 |
+
self.flow = ResidualCouplingBlock(
|
619 |
+
inter_channels, hidden_channels, 5, 1, 3, gin_channels=gin_channels
|
620 |
+
)
|
621 |
+
self.emb_g = nn.Embedding(self.spk_embed_dim, gin_channels)
|
622 |
+
self.speaker_map = None
|
623 |
+
print("gin_channels:", gin_channels, "self.spk_embed_dim:", self.spk_embed_dim)
|
624 |
+
|
625 |
+
def remove_weight_norm(self):
|
626 |
+
self.dec.remove_weight_norm()
|
627 |
+
self.flow.remove_weight_norm()
|
628 |
+
self.enc_q.remove_weight_norm()
|
629 |
+
|
630 |
+
def construct_spkmixmap(self, n_speaker):
|
631 |
+
self.speaker_map = torch.zeros((n_speaker, 1, 1, self.gin_channels))
|
632 |
+
for i in range(n_speaker):
|
633 |
+
self.speaker_map[i] = self.emb_g(torch.LongTensor([[i]]))
|
634 |
+
self.speaker_map = self.speaker_map.unsqueeze(0)
|
635 |
+
|
636 |
+
def forward(self, phone, phone_lengths, pitch, nsff0, g, rnd, max_len=None):
|
637 |
+
if self.speaker_map is not None: # [N, S] * [S, B, 1, H]
|
638 |
+
g = g.reshape((g.shape[0], g.shape[1], 1, 1, 1)) # [N, S, B, 1, 1]
|
639 |
+
g = g * self.speaker_map # [N, S, B, 1, H]
|
640 |
+
g = torch.sum(g, dim=1) # [N, 1, B, 1, H]
|
641 |
+
g = g.transpose(0, -1).transpose(0, -2).squeeze(0) # [B, H, N]
|
642 |
+
else:
|
643 |
+
g = g.unsqueeze(0)
|
644 |
+
g = self.emb_g(g).transpose(1, 2)
|
645 |
+
|
646 |
+
m_p, logs_p, x_mask = self.enc_p(phone, pitch, phone_lengths)
|
647 |
+
z_p = (m_p + torch.exp(logs_p) * rnd) * x_mask
|
648 |
+
z = self.flow(z_p, x_mask, g=g, reverse=True)
|
649 |
+
o = self.dec((z * x_mask)[:, :, :max_len], nsff0, g=g)
|
650 |
+
return o
|
651 |
+
|
652 |
+
|
653 |
+
class MultiPeriodDiscriminator(torch.nn.Module):
|
654 |
+
def __init__(self, use_spectral_norm=False):
|
655 |
+
super(MultiPeriodDiscriminator, self).__init__()
|
656 |
+
periods = [2, 3, 5, 7, 11, 17]
|
657 |
+
# periods = [3, 5, 7, 11, 17, 23, 37]
|
658 |
+
|
659 |
+
discs = [DiscriminatorS(use_spectral_norm=use_spectral_norm)]
|
660 |
+
discs = discs + [
|
661 |
+
DiscriminatorP(i, use_spectral_norm=use_spectral_norm) for i in periods
|
662 |
+
]
|
663 |
+
self.discriminators = nn.ModuleList(discs)
|
664 |
+
|
665 |
+
def forward(self, y, y_hat):
|
666 |
+
y_d_rs = [] #
|
667 |
+
y_d_gs = []
|
668 |
+
fmap_rs = []
|
669 |
+
fmap_gs = []
|
670 |
+
for i, d in enumerate(self.discriminators):
|
671 |
+
y_d_r, fmap_r = d(y)
|
672 |
+
y_d_g, fmap_g = d(y_hat)
|
673 |
+
# for j in range(len(fmap_r)):
|
674 |
+
# print(i,j,y.shape,y_hat.shape,fmap_r[j].shape,fmap_g[j].shape)
|
675 |
+
y_d_rs.append(y_d_r)
|
676 |
+
y_d_gs.append(y_d_g)
|
677 |
+
fmap_rs.append(fmap_r)
|
678 |
+
fmap_gs.append(fmap_g)
|
679 |
+
|
680 |
+
return y_d_rs, y_d_gs, fmap_rs, fmap_gs
|
681 |
+
|
682 |
+
|
683 |
+
class MultiPeriodDiscriminatorV2(torch.nn.Module):
|
684 |
+
def __init__(self, use_spectral_norm=False):
|
685 |
+
super(MultiPeriodDiscriminatorV2, self).__init__()
|
686 |
+
# periods = [2, 3, 5, 7, 11, 17]
|
687 |
+
periods = [2, 3, 5, 7, 11, 17, 23, 37]
|
688 |
+
|
689 |
+
discs = [DiscriminatorS(use_spectral_norm=use_spectral_norm)]
|
690 |
+
discs = discs + [
|
691 |
+
DiscriminatorP(i, use_spectral_norm=use_spectral_norm) for i in periods
|
692 |
+
]
|
693 |
+
self.discriminators = nn.ModuleList(discs)
|
694 |
+
|
695 |
+
def forward(self, y, y_hat):
|
696 |
+
y_d_rs = [] #
|
697 |
+
y_d_gs = []
|
698 |
+
fmap_rs = []
|
699 |
+
fmap_gs = []
|
700 |
+
for i, d in enumerate(self.discriminators):
|
701 |
+
y_d_r, fmap_r = d(y)
|
702 |
+
y_d_g, fmap_g = d(y_hat)
|
703 |
+
# for j in range(len(fmap_r)):
|
704 |
+
# print(i,j,y.shape,y_hat.shape,fmap_r[j].shape,fmap_g[j].shape)
|
705 |
+
y_d_rs.append(y_d_r)
|
706 |
+
y_d_gs.append(y_d_g)
|
707 |
+
fmap_rs.append(fmap_r)
|
708 |
+
fmap_gs.append(fmap_g)
|
709 |
+
|
710 |
+
return y_d_rs, y_d_gs, fmap_rs, fmap_gs
|
711 |
+
|
712 |
+
|
713 |
+
class DiscriminatorS(torch.nn.Module):
|
714 |
+
def __init__(self, use_spectral_norm=False):
|
715 |
+
super(DiscriminatorS, self).__init__()
|
716 |
+
norm_f = weight_norm if use_spectral_norm == False else spectral_norm
|
717 |
+
self.convs = nn.ModuleList(
|
718 |
+
[
|
719 |
+
norm_f(Conv1d(1, 16, 15, 1, padding=7)),
|
720 |
+
norm_f(Conv1d(16, 64, 41, 4, groups=4, padding=20)),
|
721 |
+
norm_f(Conv1d(64, 256, 41, 4, groups=16, padding=20)),
|
722 |
+
norm_f(Conv1d(256, 1024, 41, 4, groups=64, padding=20)),
|
723 |
+
norm_f(Conv1d(1024, 1024, 41, 4, groups=256, padding=20)),
|
724 |
+
norm_f(Conv1d(1024, 1024, 5, 1, padding=2)),
|
725 |
+
]
|
726 |
+
)
|
727 |
+
self.conv_post = norm_f(Conv1d(1024, 1, 3, 1, padding=1))
|
728 |
+
|
729 |
+
def forward(self, x):
|
730 |
+
fmap = []
|
731 |
+
|
732 |
+
for l in self.convs:
|
733 |
+
x = l(x)
|
734 |
+
x = F.leaky_relu(x, modules.LRELU_SLOPE)
|
735 |
+
fmap.append(x)
|
736 |
+
x = self.conv_post(x)
|
737 |
+
fmap.append(x)
|
738 |
+
x = torch.flatten(x, 1, -1)
|
739 |
+
|
740 |
+
return x, fmap
|
741 |
+
|
742 |
+
|
743 |
+
class DiscriminatorP(torch.nn.Module):
|
744 |
+
def __init__(self, period, kernel_size=5, stride=3, use_spectral_norm=False):
|
745 |
+
super(DiscriminatorP, self).__init__()
|
746 |
+
self.period = period
|
747 |
+
self.use_spectral_norm = use_spectral_norm
|
748 |
+
norm_f = weight_norm if use_spectral_norm == False else spectral_norm
|
749 |
+
self.convs = nn.ModuleList(
|
750 |
+
[
|
751 |
+
norm_f(
|
752 |
+
Conv2d(
|
753 |
+
1,
|
754 |
+
32,
|
755 |
+
(kernel_size, 1),
|
756 |
+
(stride, 1),
|
757 |
+
padding=(get_padding(kernel_size, 1), 0),
|
758 |
+
)
|
759 |
+
),
|
760 |
+
norm_f(
|
761 |
+
Conv2d(
|
762 |
+
32,
|
763 |
+
128,
|
764 |
+
(kernel_size, 1),
|
765 |
+
(stride, 1),
|
766 |
+
padding=(get_padding(kernel_size, 1), 0),
|
767 |
+
)
|
768 |
+
),
|
769 |
+
norm_f(
|
770 |
+
Conv2d(
|
771 |
+
128,
|
772 |
+
512,
|
773 |
+
(kernel_size, 1),
|
774 |
+
(stride, 1),
|
775 |
+
padding=(get_padding(kernel_size, 1), 0),
|
776 |
+
)
|
777 |
+
),
|
778 |
+
norm_f(
|
779 |
+
Conv2d(
|
780 |
+
512,
|
781 |
+
1024,
|
782 |
+
(kernel_size, 1),
|
783 |
+
(stride, 1),
|
784 |
+
padding=(get_padding(kernel_size, 1), 0),
|
785 |
+
)
|
786 |
+
),
|
787 |
+
norm_f(
|
788 |
+
Conv2d(
|
789 |
+
1024,
|
790 |
+
1024,
|
791 |
+
(kernel_size, 1),
|
792 |
+
1,
|
793 |
+
padding=(get_padding(kernel_size, 1), 0),
|
794 |
+
)
|
795 |
+
),
|
796 |
+
]
|
797 |
+
)
|
798 |
+
self.conv_post = norm_f(Conv2d(1024, 1, (3, 1), 1, padding=(1, 0)))
|
799 |
+
|
800 |
+
def forward(self, x):
|
801 |
+
fmap = []
|
802 |
+
|
803 |
+
# 1d to 2d
|
804 |
+
b, c, t = x.shape
|
805 |
+
if t % self.period != 0: # pad first
|
806 |
+
n_pad = self.period - (t % self.period)
|
807 |
+
x = F.pad(x, (0, n_pad), "reflect")
|
808 |
+
t = t + n_pad
|
809 |
+
x = x.view(b, c, t // self.period, self.period)
|
810 |
+
|
811 |
+
for l in self.convs:
|
812 |
+
x = l(x)
|
813 |
+
x = F.leaky_relu(x, modules.LRELU_SLOPE)
|
814 |
+
fmap.append(x)
|
815 |
+
x = self.conv_post(x)
|
816 |
+
fmap.append(x)
|
817 |
+
x = torch.flatten(x, 1, -1)
|
818 |
+
|
819 |
+
return x, fmap
|
lib/infer_pack/modules.py
ADDED
@@ -0,0 +1,522 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import copy
|
2 |
+
import math
|
3 |
+
import numpy as np
|
4 |
+
import scipy
|
5 |
+
import torch
|
6 |
+
from torch import nn
|
7 |
+
from torch.nn import functional as F
|
8 |
+
|
9 |
+
from torch.nn import Conv1d, ConvTranspose1d, AvgPool1d, Conv2d
|
10 |
+
from torch.nn.utils import weight_norm, remove_weight_norm
|
11 |
+
|
12 |
+
from lib.infer_pack import commons
|
13 |
+
from lib.infer_pack.commons import init_weights, get_padding
|
14 |
+
from lib.infer_pack.transforms import piecewise_rational_quadratic_transform
|
15 |
+
|
16 |
+
|
17 |
+
LRELU_SLOPE = 0.1
|
18 |
+
|
19 |
+
|
20 |
+
class LayerNorm(nn.Module):
|
21 |
+
def __init__(self, channels, eps=1e-5):
|
22 |
+
super().__init__()
|
23 |
+
self.channels = channels
|
24 |
+
self.eps = eps
|
25 |
+
|
26 |
+
self.gamma = nn.Parameter(torch.ones(channels))
|
27 |
+
self.beta = nn.Parameter(torch.zeros(channels))
|
28 |
+
|
29 |
+
def forward(self, x):
|
30 |
+
x = x.transpose(1, -1)
|
31 |
+
x = F.layer_norm(x, (self.channels,), self.gamma, self.beta, self.eps)
|
32 |
+
return x.transpose(1, -1)
|
33 |
+
|
34 |
+
|
35 |
+
class ConvReluNorm(nn.Module):
|
36 |
+
def __init__(
|
37 |
+
self,
|
38 |
+
in_channels,
|
39 |
+
hidden_channels,
|
40 |
+
out_channels,
|
41 |
+
kernel_size,
|
42 |
+
n_layers,
|
43 |
+
p_dropout,
|
44 |
+
):
|
45 |
+
super().__init__()
|
46 |
+
self.in_channels = in_channels
|
47 |
+
self.hidden_channels = hidden_channels
|
48 |
+
self.out_channels = out_channels
|
49 |
+
self.kernel_size = kernel_size
|
50 |
+
self.n_layers = n_layers
|
51 |
+
self.p_dropout = p_dropout
|
52 |
+
assert n_layers > 1, "Number of layers should be larger than 0."
|
53 |
+
|
54 |
+
self.conv_layers = nn.ModuleList()
|
55 |
+
self.norm_layers = nn.ModuleList()
|
56 |
+
self.conv_layers.append(
|
57 |
+
nn.Conv1d(
|
58 |
+
in_channels, hidden_channels, kernel_size, padding=kernel_size // 2
|
59 |
+
)
|
60 |
+
)
|
61 |
+
self.norm_layers.append(LayerNorm(hidden_channels))
|
62 |
+
self.relu_drop = nn.Sequential(nn.ReLU(), nn.Dropout(p_dropout))
|
63 |
+
for _ in range(n_layers - 1):
|
64 |
+
self.conv_layers.append(
|
65 |
+
nn.Conv1d(
|
66 |
+
hidden_channels,
|
67 |
+
hidden_channels,
|
68 |
+
kernel_size,
|
69 |
+
padding=kernel_size // 2,
|
70 |
+
)
|
71 |
+
)
|
72 |
+
self.norm_layers.append(LayerNorm(hidden_channels))
|
73 |
+
self.proj = nn.Conv1d(hidden_channels, out_channels, 1)
|
74 |
+
self.proj.weight.data.zero_()
|
75 |
+
self.proj.bias.data.zero_()
|
76 |
+
|
77 |
+
def forward(self, x, x_mask):
|
78 |
+
x_org = x
|
79 |
+
for i in range(self.n_layers):
|
80 |
+
x = self.conv_layers[i](x * x_mask)
|
81 |
+
x = self.norm_layers[i](x)
|
82 |
+
x = self.relu_drop(x)
|
83 |
+
x = x_org + self.proj(x)
|
84 |
+
return x * x_mask
|
85 |
+
|
86 |
+
|
87 |
+
class DDSConv(nn.Module):
|
88 |
+
"""
|
89 |
+
Dialted and Depth-Separable Convolution
|
90 |
+
"""
|
91 |
+
|
92 |
+
def __init__(self, channels, kernel_size, n_layers, p_dropout=0.0):
|
93 |
+
super().__init__()
|
94 |
+
self.channels = channels
|
95 |
+
self.kernel_size = kernel_size
|
96 |
+
self.n_layers = n_layers
|
97 |
+
self.p_dropout = p_dropout
|
98 |
+
|
99 |
+
self.drop = nn.Dropout(p_dropout)
|
100 |
+
self.convs_sep = nn.ModuleList()
|
101 |
+
self.convs_1x1 = nn.ModuleList()
|
102 |
+
self.norms_1 = nn.ModuleList()
|
103 |
+
self.norms_2 = nn.ModuleList()
|
104 |
+
for i in range(n_layers):
|
105 |
+
dilation = kernel_size**i
|
106 |
+
padding = (kernel_size * dilation - dilation) // 2
|
107 |
+
self.convs_sep.append(
|
108 |
+
nn.Conv1d(
|
109 |
+
channels,
|
110 |
+
channels,
|
111 |
+
kernel_size,
|
112 |
+
groups=channels,
|
113 |
+
dilation=dilation,
|
114 |
+
padding=padding,
|
115 |
+
)
|
116 |
+
)
|
117 |
+
self.convs_1x1.append(nn.Conv1d(channels, channels, 1))
|
118 |
+
self.norms_1.append(LayerNorm(channels))
|
119 |
+
self.norms_2.append(LayerNorm(channels))
|
120 |
+
|
121 |
+
def forward(self, x, x_mask, g=None):
|
122 |
+
if g is not None:
|
123 |
+
x = x + g
|
124 |
+
for i in range(self.n_layers):
|
125 |
+
y = self.convs_sep[i](x * x_mask)
|
126 |
+
y = self.norms_1[i](y)
|
127 |
+
y = F.gelu(y)
|
128 |
+
y = self.convs_1x1[i](y)
|
129 |
+
y = self.norms_2[i](y)
|
130 |
+
y = F.gelu(y)
|
131 |
+
y = self.drop(y)
|
132 |
+
x = x + y
|
133 |
+
return x * x_mask
|
134 |
+
|
135 |
+
|
136 |
+
class WN(torch.nn.Module):
|
137 |
+
def __init__(
|
138 |
+
self,
|
139 |
+
hidden_channels,
|
140 |
+
kernel_size,
|
141 |
+
dilation_rate,
|
142 |
+
n_layers,
|
143 |
+
gin_channels=0,
|
144 |
+
p_dropout=0,
|
145 |
+
):
|
146 |
+
super(WN, self).__init__()
|
147 |
+
assert kernel_size % 2 == 1
|
148 |
+
self.hidden_channels = hidden_channels
|
149 |
+
self.kernel_size = (kernel_size,)
|
150 |
+
self.dilation_rate = dilation_rate
|
151 |
+
self.n_layers = n_layers
|
152 |
+
self.gin_channels = gin_channels
|
153 |
+
self.p_dropout = p_dropout
|
154 |
+
|
155 |
+
self.in_layers = torch.nn.ModuleList()
|
156 |
+
self.res_skip_layers = torch.nn.ModuleList()
|
157 |
+
self.drop = nn.Dropout(p_dropout)
|
158 |
+
|
159 |
+
if gin_channels != 0:
|
160 |
+
cond_layer = torch.nn.Conv1d(
|
161 |
+
gin_channels, 2 * hidden_channels * n_layers, 1
|
162 |
+
)
|
163 |
+
self.cond_layer = torch.nn.utils.weight_norm(cond_layer, name="weight")
|
164 |
+
|
165 |
+
for i in range(n_layers):
|
166 |
+
dilation = dilation_rate**i
|
167 |
+
padding = int((kernel_size * dilation - dilation) / 2)
|
168 |
+
in_layer = torch.nn.Conv1d(
|
169 |
+
hidden_channels,
|
170 |
+
2 * hidden_channels,
|
171 |
+
kernel_size,
|
172 |
+
dilation=dilation,
|
173 |
+
padding=padding,
|
174 |
+
)
|
175 |
+
in_layer = torch.nn.utils.weight_norm(in_layer, name="weight")
|
176 |
+
self.in_layers.append(in_layer)
|
177 |
+
|
178 |
+
# last one is not necessary
|
179 |
+
if i < n_layers - 1:
|
180 |
+
res_skip_channels = 2 * hidden_channels
|
181 |
+
else:
|
182 |
+
res_skip_channels = hidden_channels
|
183 |
+
|
184 |
+
res_skip_layer = torch.nn.Conv1d(hidden_channels, res_skip_channels, 1)
|
185 |
+
res_skip_layer = torch.nn.utils.weight_norm(res_skip_layer, name="weight")
|
186 |
+
self.res_skip_layers.append(res_skip_layer)
|
187 |
+
|
188 |
+
def forward(self, x, x_mask, g=None, **kwargs):
|
189 |
+
output = torch.zeros_like(x)
|
190 |
+
n_channels_tensor = torch.IntTensor([self.hidden_channels])
|
191 |
+
|
192 |
+
if g is not None:
|
193 |
+
g = self.cond_layer(g)
|
194 |
+
|
195 |
+
for i in range(self.n_layers):
|
196 |
+
x_in = self.in_layers[i](x)
|
197 |
+
if g is not None:
|
198 |
+
cond_offset = i * 2 * self.hidden_channels
|
199 |
+
g_l = g[:, cond_offset : cond_offset + 2 * self.hidden_channels, :]
|
200 |
+
else:
|
201 |
+
g_l = torch.zeros_like(x_in)
|
202 |
+
|
203 |
+
acts = commons.fused_add_tanh_sigmoid_multiply(x_in, g_l, n_channels_tensor)
|
204 |
+
acts = self.drop(acts)
|
205 |
+
|
206 |
+
res_skip_acts = self.res_skip_layers[i](acts)
|
207 |
+
if i < self.n_layers - 1:
|
208 |
+
res_acts = res_skip_acts[:, : self.hidden_channels, :]
|
209 |
+
x = (x + res_acts) * x_mask
|
210 |
+
output = output + res_skip_acts[:, self.hidden_channels :, :]
|
211 |
+
else:
|
212 |
+
output = output + res_skip_acts
|
213 |
+
return output * x_mask
|
214 |
+
|
215 |
+
def remove_weight_norm(self):
|
216 |
+
if self.gin_channels != 0:
|
217 |
+
torch.nn.utils.remove_weight_norm(self.cond_layer)
|
218 |
+
for l in self.in_layers:
|
219 |
+
torch.nn.utils.remove_weight_norm(l)
|
220 |
+
for l in self.res_skip_layers:
|
221 |
+
torch.nn.utils.remove_weight_norm(l)
|
222 |
+
|
223 |
+
|
224 |
+
class ResBlock1(torch.nn.Module):
|
225 |
+
def __init__(self, channels, kernel_size=3, dilation=(1, 3, 5)):
|
226 |
+
super(ResBlock1, self).__init__()
|
227 |
+
self.convs1 = nn.ModuleList(
|
228 |
+
[
|
229 |
+
weight_norm(
|
230 |
+
Conv1d(
|
231 |
+
channels,
|
232 |
+
channels,
|
233 |
+
kernel_size,
|
234 |
+
1,
|
235 |
+
dilation=dilation[0],
|
236 |
+
padding=get_padding(kernel_size, dilation[0]),
|
237 |
+
)
|
238 |
+
),
|
239 |
+
weight_norm(
|
240 |
+
Conv1d(
|
241 |
+
channels,
|
242 |
+
channels,
|
243 |
+
kernel_size,
|
244 |
+
1,
|
245 |
+
dilation=dilation[1],
|
246 |
+
padding=get_padding(kernel_size, dilation[1]),
|
247 |
+
)
|
248 |
+
),
|
249 |
+
weight_norm(
|
250 |
+
Conv1d(
|
251 |
+
channels,
|
252 |
+
channels,
|
253 |
+
kernel_size,
|
254 |
+
1,
|
255 |
+
dilation=dilation[2],
|
256 |
+
padding=get_padding(kernel_size, dilation[2]),
|
257 |
+
)
|
258 |
+
),
|
259 |
+
]
|
260 |
+
)
|
261 |
+
self.convs1.apply(init_weights)
|
262 |
+
|
263 |
+
self.convs2 = nn.ModuleList(
|
264 |
+
[
|
265 |
+
weight_norm(
|
266 |
+
Conv1d(
|
267 |
+
channels,
|
268 |
+
channels,
|
269 |
+
kernel_size,
|
270 |
+
1,
|
271 |
+
dilation=1,
|
272 |
+
padding=get_padding(kernel_size, 1),
|
273 |
+
)
|
274 |
+
),
|
275 |
+
weight_norm(
|
276 |
+
Conv1d(
|
277 |
+
channels,
|
278 |
+
channels,
|
279 |
+
kernel_size,
|
280 |
+
1,
|
281 |
+
dilation=1,
|
282 |
+
padding=get_padding(kernel_size, 1),
|
283 |
+
)
|
284 |
+
),
|
285 |
+
weight_norm(
|
286 |
+
Conv1d(
|
287 |
+
channels,
|
288 |
+
channels,
|
289 |
+
kernel_size,
|
290 |
+
1,
|
291 |
+
dilation=1,
|
292 |
+
padding=get_padding(kernel_size, 1),
|
293 |
+
)
|
294 |
+
),
|
295 |
+
]
|
296 |
+
)
|
297 |
+
self.convs2.apply(init_weights)
|
298 |
+
|
299 |
+
def forward(self, x, x_mask=None):
|
300 |
+
for c1, c2 in zip(self.convs1, self.convs2):
|
301 |
+
xt = F.leaky_relu(x, LRELU_SLOPE)
|
302 |
+
if x_mask is not None:
|
303 |
+
xt = xt * x_mask
|
304 |
+
xt = c1(xt)
|
305 |
+
xt = F.leaky_relu(xt, LRELU_SLOPE)
|
306 |
+
if x_mask is not None:
|
307 |
+
xt = xt * x_mask
|
308 |
+
xt = c2(xt)
|
309 |
+
x = xt + x
|
310 |
+
if x_mask is not None:
|
311 |
+
x = x * x_mask
|
312 |
+
return x
|
313 |
+
|
314 |
+
def remove_weight_norm(self):
|
315 |
+
for l in self.convs1:
|
316 |
+
remove_weight_norm(l)
|
317 |
+
for l in self.convs2:
|
318 |
+
remove_weight_norm(l)
|
319 |
+
|
320 |
+
|
321 |
+
class ResBlock2(torch.nn.Module):
|
322 |
+
def __init__(self, channels, kernel_size=3, dilation=(1, 3)):
|
323 |
+
super(ResBlock2, self).__init__()
|
324 |
+
self.convs = nn.ModuleList(
|
325 |
+
[
|
326 |
+
weight_norm(
|
327 |
+
Conv1d(
|
328 |
+
channels,
|
329 |
+
channels,
|
330 |
+
kernel_size,
|
331 |
+
1,
|
332 |
+
dilation=dilation[0],
|
333 |
+
padding=get_padding(kernel_size, dilation[0]),
|
334 |
+
)
|
335 |
+
),
|
336 |
+
weight_norm(
|
337 |
+
Conv1d(
|
338 |
+
channels,
|
339 |
+
channels,
|
340 |
+
kernel_size,
|
341 |
+
1,
|
342 |
+
dilation=dilation[1],
|
343 |
+
padding=get_padding(kernel_size, dilation[1]),
|
344 |
+
)
|
345 |
+
),
|
346 |
+
]
|
347 |
+
)
|
348 |
+
self.convs.apply(init_weights)
|
349 |
+
|
350 |
+
def forward(self, x, x_mask=None):
|
351 |
+
for c in self.convs:
|
352 |
+
xt = F.leaky_relu(x, LRELU_SLOPE)
|
353 |
+
if x_mask is not None:
|
354 |
+
xt = xt * x_mask
|
355 |
+
xt = c(xt)
|
356 |
+
x = xt + x
|
357 |
+
if x_mask is not None:
|
358 |
+
x = x * x_mask
|
359 |
+
return x
|
360 |
+
|
361 |
+
def remove_weight_norm(self):
|
362 |
+
for l in self.convs:
|
363 |
+
remove_weight_norm(l)
|
364 |
+
|
365 |
+
|
366 |
+
class Log(nn.Module):
|
367 |
+
def forward(self, x, x_mask, reverse=False, **kwargs):
|
368 |
+
if not reverse:
|
369 |
+
y = torch.log(torch.clamp_min(x, 1e-5)) * x_mask
|
370 |
+
logdet = torch.sum(-y, [1, 2])
|
371 |
+
return y, logdet
|
372 |
+
else:
|
373 |
+
x = torch.exp(x) * x_mask
|
374 |
+
return x
|
375 |
+
|
376 |
+
|
377 |
+
class Flip(nn.Module):
|
378 |
+
def forward(self, x, *args, reverse=False, **kwargs):
|
379 |
+
x = torch.flip(x, [1])
|
380 |
+
if not reverse:
|
381 |
+
logdet = torch.zeros(x.size(0)).to(dtype=x.dtype, device=x.device)
|
382 |
+
return x, logdet
|
383 |
+
else:
|
384 |
+
return x
|
385 |
+
|
386 |
+
|
387 |
+
class ElementwiseAffine(nn.Module):
|
388 |
+
def __init__(self, channels):
|
389 |
+
super().__init__()
|
390 |
+
self.channels = channels
|
391 |
+
self.m = nn.Parameter(torch.zeros(channels, 1))
|
392 |
+
self.logs = nn.Parameter(torch.zeros(channels, 1))
|
393 |
+
|
394 |
+
def forward(self, x, x_mask, reverse=False, **kwargs):
|
395 |
+
if not reverse:
|
396 |
+
y = self.m + torch.exp(self.logs) * x
|
397 |
+
y = y * x_mask
|
398 |
+
logdet = torch.sum(self.logs * x_mask, [1, 2])
|
399 |
+
return y, logdet
|
400 |
+
else:
|
401 |
+
x = (x - self.m) * torch.exp(-self.logs) * x_mask
|
402 |
+
return x
|
403 |
+
|
404 |
+
|
405 |
+
class ResidualCouplingLayer(nn.Module):
|
406 |
+
def __init__(
|
407 |
+
self,
|
408 |
+
channels,
|
409 |
+
hidden_channels,
|
410 |
+
kernel_size,
|
411 |
+
dilation_rate,
|
412 |
+
n_layers,
|
413 |
+
p_dropout=0,
|
414 |
+
gin_channels=0,
|
415 |
+
mean_only=False,
|
416 |
+
):
|
417 |
+
assert channels % 2 == 0, "channels should be divisible by 2"
|
418 |
+
super().__init__()
|
419 |
+
self.channels = channels
|
420 |
+
self.hidden_channels = hidden_channels
|
421 |
+
self.kernel_size = kernel_size
|
422 |
+
self.dilation_rate = dilation_rate
|
423 |
+
self.n_layers = n_layers
|
424 |
+
self.half_channels = channels // 2
|
425 |
+
self.mean_only = mean_only
|
426 |
+
|
427 |
+
self.pre = nn.Conv1d(self.half_channels, hidden_channels, 1)
|
428 |
+
self.enc = WN(
|
429 |
+
hidden_channels,
|
430 |
+
kernel_size,
|
431 |
+
dilation_rate,
|
432 |
+
n_layers,
|
433 |
+
p_dropout=p_dropout,
|
434 |
+
gin_channels=gin_channels,
|
435 |
+
)
|
436 |
+
self.post = nn.Conv1d(hidden_channels, self.half_channels * (2 - mean_only), 1)
|
437 |
+
self.post.weight.data.zero_()
|
438 |
+
self.post.bias.data.zero_()
|
439 |
+
|
440 |
+
def forward(self, x, x_mask, g=None, reverse=False):
|
441 |
+
x0, x1 = torch.split(x, [self.half_channels] * 2, 1)
|
442 |
+
h = self.pre(x0) * x_mask
|
443 |
+
h = self.enc(h, x_mask, g=g)
|
444 |
+
stats = self.post(h) * x_mask
|
445 |
+
if not self.mean_only:
|
446 |
+
m, logs = torch.split(stats, [self.half_channels] * 2, 1)
|
447 |
+
else:
|
448 |
+
m = stats
|
449 |
+
logs = torch.zeros_like(m)
|
450 |
+
|
451 |
+
if not reverse:
|
452 |
+
x1 = m + x1 * torch.exp(logs) * x_mask
|
453 |
+
x = torch.cat([x0, x1], 1)
|
454 |
+
logdet = torch.sum(logs, [1, 2])
|
455 |
+
return x, logdet
|
456 |
+
else:
|
457 |
+
x1 = (x1 - m) * torch.exp(-logs) * x_mask
|
458 |
+
x = torch.cat([x0, x1], 1)
|
459 |
+
return x
|
460 |
+
|
461 |
+
def remove_weight_norm(self):
|
462 |
+
self.enc.remove_weight_norm()
|
463 |
+
|
464 |
+
|
465 |
+
class ConvFlow(nn.Module):
|
466 |
+
def __init__(
|
467 |
+
self,
|
468 |
+
in_channels,
|
469 |
+
filter_channels,
|
470 |
+
kernel_size,
|
471 |
+
n_layers,
|
472 |
+
num_bins=10,
|
473 |
+
tail_bound=5.0,
|
474 |
+
):
|
475 |
+
super().__init__()
|
476 |
+
self.in_channels = in_channels
|
477 |
+
self.filter_channels = filter_channels
|
478 |
+
self.kernel_size = kernel_size
|
479 |
+
self.n_layers = n_layers
|
480 |
+
self.num_bins = num_bins
|
481 |
+
self.tail_bound = tail_bound
|
482 |
+
self.half_channels = in_channels // 2
|
483 |
+
|
484 |
+
self.pre = nn.Conv1d(self.half_channels, filter_channels, 1)
|
485 |
+
self.convs = DDSConv(filter_channels, kernel_size, n_layers, p_dropout=0.0)
|
486 |
+
self.proj = nn.Conv1d(
|
487 |
+
filter_channels, self.half_channels * (num_bins * 3 - 1), 1
|
488 |
+
)
|
489 |
+
self.proj.weight.data.zero_()
|
490 |
+
self.proj.bias.data.zero_()
|
491 |
+
|
492 |
+
def forward(self, x, x_mask, g=None, reverse=False):
|
493 |
+
x0, x1 = torch.split(x, [self.half_channels] * 2, 1)
|
494 |
+
h = self.pre(x0)
|
495 |
+
h = self.convs(h, x_mask, g=g)
|
496 |
+
h = self.proj(h) * x_mask
|
497 |
+
|
498 |
+
b, c, t = x0.shape
|
499 |
+
h = h.reshape(b, c, -1, t).permute(0, 1, 3, 2) # [b, cx?, t] -> [b, c, t, ?]
|
500 |
+
|
501 |
+
unnormalized_widths = h[..., : self.num_bins] / math.sqrt(self.filter_channels)
|
502 |
+
unnormalized_heights = h[..., self.num_bins : 2 * self.num_bins] / math.sqrt(
|
503 |
+
self.filter_channels
|
504 |
+
)
|
505 |
+
unnormalized_derivatives = h[..., 2 * self.num_bins :]
|
506 |
+
|
507 |
+
x1, logabsdet = piecewise_rational_quadratic_transform(
|
508 |
+
x1,
|
509 |
+
unnormalized_widths,
|
510 |
+
unnormalized_heights,
|
511 |
+
unnormalized_derivatives,
|
512 |
+
inverse=reverse,
|
513 |
+
tails="linear",
|
514 |
+
tail_bound=self.tail_bound,
|
515 |
+
)
|
516 |
+
|
517 |
+
x = torch.cat([x0, x1], 1) * x_mask
|
518 |
+
logdet = torch.sum(logabsdet * x_mask, [1, 2])
|
519 |
+
if not reverse:
|
520 |
+
return x, logdet
|
521 |
+
else:
|
522 |
+
return x
|
lib/infer_pack/modules/F0Predictor/DioF0Predictor.py
ADDED
@@ -0,0 +1,90 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from lib.infer_pack.modules.F0Predictor.F0Predictor import F0Predictor
|
2 |
+
import pyworld
|
3 |
+
import numpy as np
|
4 |
+
|
5 |
+
|
6 |
+
class DioF0Predictor(F0Predictor):
|
7 |
+
def __init__(self, hop_length=512, f0_min=50, f0_max=1100, sampling_rate=44100):
|
8 |
+
self.hop_length = hop_length
|
9 |
+
self.f0_min = f0_min
|
10 |
+
self.f0_max = f0_max
|
11 |
+
self.sampling_rate = sampling_rate
|
12 |
+
|
13 |
+
def interpolate_f0(self, f0):
|
14 |
+
"""
|
15 |
+
对F0进行插值处理
|
16 |
+
"""
|
17 |
+
|
18 |
+
data = np.reshape(f0, (f0.size, 1))
|
19 |
+
|
20 |
+
vuv_vector = np.zeros((data.size, 1), dtype=np.float32)
|
21 |
+
vuv_vector[data > 0.0] = 1.0
|
22 |
+
vuv_vector[data <= 0.0] = 0.0
|
23 |
+
|
24 |
+
ip_data = data
|
25 |
+
|
26 |
+
frame_number = data.size
|
27 |
+
last_value = 0.0
|
28 |
+
for i in range(frame_number):
|
29 |
+
if data[i] <= 0.0:
|
30 |
+
j = i + 1
|
31 |
+
for j in range(i + 1, frame_number):
|
32 |
+
if data[j] > 0.0:
|
33 |
+
break
|
34 |
+
if j < frame_number - 1:
|
35 |
+
if last_value > 0.0:
|
36 |
+
step = (data[j] - data[i - 1]) / float(j - i)
|
37 |
+
for k in range(i, j):
|
38 |
+
ip_data[k] = data[i - 1] + step * (k - i + 1)
|
39 |
+
else:
|
40 |
+
for k in range(i, j):
|
41 |
+
ip_data[k] = data[j]
|
42 |
+
else:
|
43 |
+
for k in range(i, frame_number):
|
44 |
+
ip_data[k] = last_value
|
45 |
+
else:
|
46 |
+
ip_data[i] = data[i] # 这里可能存在一个没有必要的拷贝
|
47 |
+
last_value = data[i]
|
48 |
+
|
49 |
+
return ip_data[:, 0], vuv_vector[:, 0]
|
50 |
+
|
51 |
+
def resize_f0(self, x, target_len):
|
52 |
+
source = np.array(x)
|
53 |
+
source[source < 0.001] = np.nan
|
54 |
+
target = np.interp(
|
55 |
+
np.arange(0, len(source) * target_len, len(source)) / target_len,
|
56 |
+
np.arange(0, len(source)),
|
57 |
+
source,
|
58 |
+
)
|
59 |
+
res = np.nan_to_num(target)
|
60 |
+
return res
|
61 |
+
|
62 |
+
def compute_f0(self, wav, p_len=None):
|
63 |
+
if p_len is None:
|
64 |
+
p_len = wav.shape[0] // self.hop_length
|
65 |
+
f0, t = pyworld.dio(
|
66 |
+
wav.astype(np.double),
|
67 |
+
fs=self.sampling_rate,
|
68 |
+
f0_floor=self.f0_min,
|
69 |
+
f0_ceil=self.f0_max,
|
70 |
+
frame_period=1000 * self.hop_length / self.sampling_rate,
|
71 |
+
)
|
72 |
+
f0 = pyworld.stonemask(wav.astype(np.double), f0, t, self.sampling_rate)
|
73 |
+
for index, pitch in enumerate(f0):
|
74 |
+
f0[index] = round(pitch, 1)
|
75 |
+
return self.interpolate_f0(self.resize_f0(f0, p_len))[0]
|
76 |
+
|
77 |
+
def compute_f0_uv(self, wav, p_len=None):
|
78 |
+
if p_len is None:
|
79 |
+
p_len = wav.shape[0] // self.hop_length
|
80 |
+
f0, t = pyworld.dio(
|
81 |
+
wav.astype(np.double),
|
82 |
+
fs=self.sampling_rate,
|
83 |
+
f0_floor=self.f0_min,
|
84 |
+
f0_ceil=self.f0_max,
|
85 |
+
frame_period=1000 * self.hop_length / self.sampling_rate,
|
86 |
+
)
|
87 |
+
f0 = pyworld.stonemask(wav.astype(np.double), f0, t, self.sampling_rate)
|
88 |
+
for index, pitch in enumerate(f0):
|
89 |
+
f0[index] = round(pitch, 1)
|
90 |
+
return self.interpolate_f0(self.resize_f0(f0, p_len))
|
lib/infer_pack/modules/F0Predictor/F0Predictor.py
ADDED
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
class F0Predictor(object):
|
2 |
+
def compute_f0(self, wav, p_len):
|
3 |
+
"""
|
4 |
+
input: wav:[signal_length]
|
5 |
+
p_len:int
|
6 |
+
output: f0:[signal_length//hop_length]
|
7 |
+
"""
|
8 |
+
pass
|
9 |
+
|
10 |
+
def compute_f0_uv(self, wav, p_len):
|
11 |
+
"""
|
12 |
+
input: wav:[signal_length]
|
13 |
+
p_len:int
|
14 |
+
output: f0:[signal_length//hop_length],uv:[signal_length//hop_length]
|
15 |
+
"""
|
16 |
+
pass
|
lib/infer_pack/modules/F0Predictor/HarvestF0Predictor.py
ADDED
@@ -0,0 +1,86 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from lib.infer_pack.modules.F0Predictor.F0Predictor import F0Predictor
|
2 |
+
import pyworld
|
3 |
+
import numpy as np
|
4 |
+
|
5 |
+
|
6 |
+
class HarvestF0Predictor(F0Predictor):
|
7 |
+
def __init__(self, hop_length=512, f0_min=50, f0_max=1100, sampling_rate=44100):
|
8 |
+
self.hop_length = hop_length
|
9 |
+
self.f0_min = f0_min
|
10 |
+
self.f0_max = f0_max
|
11 |
+
self.sampling_rate = sampling_rate
|
12 |
+
|
13 |
+
def interpolate_f0(self, f0):
|
14 |
+
"""
|
15 |
+
对F0进行插值处理
|
16 |
+
"""
|
17 |
+
|
18 |
+
data = np.reshape(f0, (f0.size, 1))
|
19 |
+
|
20 |
+
vuv_vector = np.zeros((data.size, 1), dtype=np.float32)
|
21 |
+
vuv_vector[data > 0.0] = 1.0
|
22 |
+
vuv_vector[data <= 0.0] = 0.0
|
23 |
+
|
24 |
+
ip_data = data
|
25 |
+
|
26 |
+
frame_number = data.size
|
27 |
+
last_value = 0.0
|
28 |
+
for i in range(frame_number):
|
29 |
+
if data[i] <= 0.0:
|
30 |
+
j = i + 1
|
31 |
+
for j in range(i + 1, frame_number):
|
32 |
+
if data[j] > 0.0:
|
33 |
+
break
|
34 |
+
if j < frame_number - 1:
|
35 |
+
if last_value > 0.0:
|
36 |
+
step = (data[j] - data[i - 1]) / float(j - i)
|
37 |
+
for k in range(i, j):
|
38 |
+
ip_data[k] = data[i - 1] + step * (k - i + 1)
|
39 |
+
else:
|
40 |
+
for k in range(i, j):
|
41 |
+
ip_data[k] = data[j]
|
42 |
+
else:
|
43 |
+
for k in range(i, frame_number):
|
44 |
+
ip_data[k] = last_value
|
45 |
+
else:
|
46 |
+
ip_data[i] = data[i] # 这里可能存在一个没有必要的拷贝
|
47 |
+
last_value = data[i]
|
48 |
+
|
49 |
+
return ip_data[:, 0], vuv_vector[:, 0]
|
50 |
+
|
51 |
+
def resize_f0(self, x, target_len):
|
52 |
+
source = np.array(x)
|
53 |
+
source[source < 0.001] = np.nan
|
54 |
+
target = np.interp(
|
55 |
+
np.arange(0, len(source) * target_len, len(source)) / target_len,
|
56 |
+
np.arange(0, len(source)),
|
57 |
+
source,
|
58 |
+
)
|
59 |
+
res = np.nan_to_num(target)
|
60 |
+
return res
|
61 |
+
|
62 |
+
def compute_f0(self, wav, p_len=None):
|
63 |
+
if p_len is None:
|
64 |
+
p_len = wav.shape[0] // self.hop_length
|
65 |
+
f0, t = pyworld.harvest(
|
66 |
+
wav.astype(np.double),
|
67 |
+
fs=self.hop_length,
|
68 |
+
f0_ceil=self.f0_max,
|
69 |
+
f0_floor=self.f0_min,
|
70 |
+
frame_period=1000 * self.hop_length / self.sampling_rate,
|
71 |
+
)
|
72 |
+
f0 = pyworld.stonemask(wav.astype(np.double), f0, t, self.fs)
|
73 |
+
return self.interpolate_f0(self.resize_f0(f0, p_len))[0]
|
74 |
+
|
75 |
+
def compute_f0_uv(self, wav, p_len=None):
|
76 |
+
if p_len is None:
|
77 |
+
p_len = wav.shape[0] // self.hop_length
|
78 |
+
f0, t = pyworld.harvest(
|
79 |
+
wav.astype(np.double),
|
80 |
+
fs=self.sampling_rate,
|
81 |
+
f0_floor=self.f0_min,
|
82 |
+
f0_ceil=self.f0_max,
|
83 |
+
frame_period=1000 * self.hop_length / self.sampling_rate,
|
84 |
+
)
|
85 |
+
f0 = pyworld.stonemask(wav.astype(np.double), f0, t, self.sampling_rate)
|
86 |
+
return self.interpolate_f0(self.resize_f0(f0, p_len))
|
lib/infer_pack/modules/F0Predictor/PMF0Predictor.py
ADDED
@@ -0,0 +1,97 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from lib.infer_pack.modules.F0Predictor.F0Predictor import F0Predictor
|
2 |
+
import parselmouth
|
3 |
+
import numpy as np
|
4 |
+
|
5 |
+
|
6 |
+
class PMF0Predictor(F0Predictor):
|
7 |
+
def __init__(self, hop_length=512, f0_min=50, f0_max=1100, sampling_rate=44100):
|
8 |
+
self.hop_length = hop_length
|
9 |
+
self.f0_min = f0_min
|
10 |
+
self.f0_max = f0_max
|
11 |
+
self.sampling_rate = sampling_rate
|
12 |
+
|
13 |
+
def interpolate_f0(self, f0):
|
14 |
+
"""
|
15 |
+
对F0进行插值处理
|
16 |
+
"""
|
17 |
+
|
18 |
+
data = np.reshape(f0, (f0.size, 1))
|
19 |
+
|
20 |
+
vuv_vector = np.zeros((data.size, 1), dtype=np.float32)
|
21 |
+
vuv_vector[data > 0.0] = 1.0
|
22 |
+
vuv_vector[data <= 0.0] = 0.0
|
23 |
+
|
24 |
+
ip_data = data
|
25 |
+
|
26 |
+
frame_number = data.size
|
27 |
+
last_value = 0.0
|
28 |
+
for i in range(frame_number):
|
29 |
+
if data[i] <= 0.0:
|
30 |
+
j = i + 1
|
31 |
+
for j in range(i + 1, frame_number):
|
32 |
+
if data[j] > 0.0:
|
33 |
+
break
|
34 |
+
if j < frame_number - 1:
|
35 |
+
if last_value > 0.0:
|
36 |
+
step = (data[j] - data[i - 1]) / float(j - i)
|
37 |
+
for k in range(i, j):
|
38 |
+
ip_data[k] = data[i - 1] + step * (k - i + 1)
|
39 |
+
else:
|
40 |
+
for k in range(i, j):
|
41 |
+
ip_data[k] = data[j]
|
42 |
+
else:
|
43 |
+
for k in range(i, frame_number):
|
44 |
+
ip_data[k] = last_value
|
45 |
+
else:
|
46 |
+
ip_data[i] = data[i] # 这里可能存在一个没有必要的拷贝
|
47 |
+
last_value = data[i]
|
48 |
+
|
49 |
+
return ip_data[:, 0], vuv_vector[:, 0]
|
50 |
+
|
51 |
+
def compute_f0(self, wav, p_len=None):
|
52 |
+
x = wav
|
53 |
+
if p_len is None:
|
54 |
+
p_len = x.shape[0] // self.hop_length
|
55 |
+
else:
|
56 |
+
assert abs(p_len - x.shape[0] // self.hop_length) < 4, "pad length error"
|
57 |
+
time_step = self.hop_length / self.sampling_rate * 1000
|
58 |
+
f0 = (
|
59 |
+
parselmouth.Sound(x, self.sampling_rate)
|
60 |
+
.to_pitch_ac(
|
61 |
+
time_step=time_step / 1000,
|
62 |
+
voicing_threshold=0.6,
|
63 |
+
pitch_floor=self.f0_min,
|
64 |
+
pitch_ceiling=self.f0_max,
|
65 |
+
)
|
66 |
+
.selected_array["frequency"]
|
67 |
+
)
|
68 |
+
|
69 |
+
pad_size = (p_len - len(f0) + 1) // 2
|
70 |
+
if pad_size > 0 or p_len - len(f0) - pad_size > 0:
|
71 |
+
f0 = np.pad(f0, [[pad_size, p_len - len(f0) - pad_size]], mode="constant")
|
72 |
+
f0, uv = self.interpolate_f0(f0)
|
73 |
+
return f0
|
74 |
+
|
75 |
+
def compute_f0_uv(self, wav, p_len=None):
|
76 |
+
x = wav
|
77 |
+
if p_len is None:
|
78 |
+
p_len = x.shape[0] // self.hop_length
|
79 |
+
else:
|
80 |
+
assert abs(p_len - x.shape[0] // self.hop_length) < 4, "pad length error"
|
81 |
+
time_step = self.hop_length / self.sampling_rate * 1000
|
82 |
+
f0 = (
|
83 |
+
parselmouth.Sound(x, self.sampling_rate)
|
84 |
+
.to_pitch_ac(
|
85 |
+
time_step=time_step / 1000,
|
86 |
+
voicing_threshold=0.6,
|
87 |
+
pitch_floor=self.f0_min,
|
88 |
+
pitch_ceiling=self.f0_max,
|
89 |
+
)
|
90 |
+
.selected_array["frequency"]
|
91 |
+
)
|
92 |
+
|
93 |
+
pad_size = (p_len - len(f0) + 1) // 2
|
94 |
+
if pad_size > 0 or p_len - len(f0) - pad_size > 0:
|
95 |
+
f0 = np.pad(f0, [[pad_size, p_len - len(f0) - pad_size]], mode="constant")
|
96 |
+
f0, uv = self.interpolate_f0(f0)
|
97 |
+
return f0, uv
|
lib/infer_pack/modules/F0Predictor/__init__.py
ADDED
File without changes
|
lib/infer_pack/transforms.py
ADDED
@@ -0,0 +1,209 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from torch.nn import functional as F
|
3 |
+
|
4 |
+
import numpy as np
|
5 |
+
|
6 |
+
|
7 |
+
DEFAULT_MIN_BIN_WIDTH = 1e-3
|
8 |
+
DEFAULT_MIN_BIN_HEIGHT = 1e-3
|
9 |
+
DEFAULT_MIN_DERIVATIVE = 1e-3
|
10 |
+
|
11 |
+
|
12 |
+
def piecewise_rational_quadratic_transform(
|
13 |
+
inputs,
|
14 |
+
unnormalized_widths,
|
15 |
+
unnormalized_heights,
|
16 |
+
unnormalized_derivatives,
|
17 |
+
inverse=False,
|
18 |
+
tails=None,
|
19 |
+
tail_bound=1.0,
|
20 |
+
min_bin_width=DEFAULT_MIN_BIN_WIDTH,
|
21 |
+
min_bin_height=DEFAULT_MIN_BIN_HEIGHT,
|
22 |
+
min_derivative=DEFAULT_MIN_DERIVATIVE,
|
23 |
+
):
|
24 |
+
if tails is None:
|
25 |
+
spline_fn = rational_quadratic_spline
|
26 |
+
spline_kwargs = {}
|
27 |
+
else:
|
28 |
+
spline_fn = unconstrained_rational_quadratic_spline
|
29 |
+
spline_kwargs = {"tails": tails, "tail_bound": tail_bound}
|
30 |
+
|
31 |
+
outputs, logabsdet = spline_fn(
|
32 |
+
inputs=inputs,
|
33 |
+
unnormalized_widths=unnormalized_widths,
|
34 |
+
unnormalized_heights=unnormalized_heights,
|
35 |
+
unnormalized_derivatives=unnormalized_derivatives,
|
36 |
+
inverse=inverse,
|
37 |
+
min_bin_width=min_bin_width,
|
38 |
+
min_bin_height=min_bin_height,
|
39 |
+
min_derivative=min_derivative,
|
40 |
+
**spline_kwargs
|
41 |
+
)
|
42 |
+
return outputs, logabsdet
|
43 |
+
|
44 |
+
|
45 |
+
def searchsorted(bin_locations, inputs, eps=1e-6):
|
46 |
+
bin_locations[..., -1] += eps
|
47 |
+
return torch.sum(inputs[..., None] >= bin_locations, dim=-1) - 1
|
48 |
+
|
49 |
+
|
50 |
+
def unconstrained_rational_quadratic_spline(
|
51 |
+
inputs,
|
52 |
+
unnormalized_widths,
|
53 |
+
unnormalized_heights,
|
54 |
+
unnormalized_derivatives,
|
55 |
+
inverse=False,
|
56 |
+
tails="linear",
|
57 |
+
tail_bound=1.0,
|
58 |
+
min_bin_width=DEFAULT_MIN_BIN_WIDTH,
|
59 |
+
min_bin_height=DEFAULT_MIN_BIN_HEIGHT,
|
60 |
+
min_derivative=DEFAULT_MIN_DERIVATIVE,
|
61 |
+
):
|
62 |
+
inside_interval_mask = (inputs >= -tail_bound) & (inputs <= tail_bound)
|
63 |
+
outside_interval_mask = ~inside_interval_mask
|
64 |
+
|
65 |
+
outputs = torch.zeros_like(inputs)
|
66 |
+
logabsdet = torch.zeros_like(inputs)
|
67 |
+
|
68 |
+
if tails == "linear":
|
69 |
+
unnormalized_derivatives = F.pad(unnormalized_derivatives, pad=(1, 1))
|
70 |
+
constant = np.log(np.exp(1 - min_derivative) - 1)
|
71 |
+
unnormalized_derivatives[..., 0] = constant
|
72 |
+
unnormalized_derivatives[..., -1] = constant
|
73 |
+
|
74 |
+
outputs[outside_interval_mask] = inputs[outside_interval_mask]
|
75 |
+
logabsdet[outside_interval_mask] = 0
|
76 |
+
else:
|
77 |
+
raise RuntimeError("{} tails are not implemented.".format(tails))
|
78 |
+
|
79 |
+
(
|
80 |
+
outputs[inside_interval_mask],
|
81 |
+
logabsdet[inside_interval_mask],
|
82 |
+
) = rational_quadratic_spline(
|
83 |
+
inputs=inputs[inside_interval_mask],
|
84 |
+
unnormalized_widths=unnormalized_widths[inside_interval_mask, :],
|
85 |
+
unnormalized_heights=unnormalized_heights[inside_interval_mask, :],
|
86 |
+
unnormalized_derivatives=unnormalized_derivatives[inside_interval_mask, :],
|
87 |
+
inverse=inverse,
|
88 |
+
left=-tail_bound,
|
89 |
+
right=tail_bound,
|
90 |
+
bottom=-tail_bound,
|
91 |
+
top=tail_bound,
|
92 |
+
min_bin_width=min_bin_width,
|
93 |
+
min_bin_height=min_bin_height,
|
94 |
+
min_derivative=min_derivative,
|
95 |
+
)
|
96 |
+
|
97 |
+
return outputs, logabsdet
|
98 |
+
|
99 |
+
|
100 |
+
def rational_quadratic_spline(
|
101 |
+
inputs,
|
102 |
+
unnormalized_widths,
|
103 |
+
unnormalized_heights,
|
104 |
+
unnormalized_derivatives,
|
105 |
+
inverse=False,
|
106 |
+
left=0.0,
|
107 |
+
right=1.0,
|
108 |
+
bottom=0.0,
|
109 |
+
top=1.0,
|
110 |
+
min_bin_width=DEFAULT_MIN_BIN_WIDTH,
|
111 |
+
min_bin_height=DEFAULT_MIN_BIN_HEIGHT,
|
112 |
+
min_derivative=DEFAULT_MIN_DERIVATIVE,
|
113 |
+
):
|
114 |
+
if torch.min(inputs) < left or torch.max(inputs) > right:
|
115 |
+
raise ValueError("Input to a transform is not within its domain")
|
116 |
+
|
117 |
+
num_bins = unnormalized_widths.shape[-1]
|
118 |
+
|
119 |
+
if min_bin_width * num_bins > 1.0:
|
120 |
+
raise ValueError("Minimal bin width too large for the number of bins")
|
121 |
+
if min_bin_height * num_bins > 1.0:
|
122 |
+
raise ValueError("Minimal bin height too large for the number of bins")
|
123 |
+
|
124 |
+
widths = F.softmax(unnormalized_widths, dim=-1)
|
125 |
+
widths = min_bin_width + (1 - min_bin_width * num_bins) * widths
|
126 |
+
cumwidths = torch.cumsum(widths, dim=-1)
|
127 |
+
cumwidths = F.pad(cumwidths, pad=(1, 0), mode="constant", value=0.0)
|
128 |
+
cumwidths = (right - left) * cumwidths + left
|
129 |
+
cumwidths[..., 0] = left
|
130 |
+
cumwidths[..., -1] = right
|
131 |
+
widths = cumwidths[..., 1:] - cumwidths[..., :-1]
|
132 |
+
|
133 |
+
derivatives = min_derivative + F.softplus(unnormalized_derivatives)
|
134 |
+
|
135 |
+
heights = F.softmax(unnormalized_heights, dim=-1)
|
136 |
+
heights = min_bin_height + (1 - min_bin_height * num_bins) * heights
|
137 |
+
cumheights = torch.cumsum(heights, dim=-1)
|
138 |
+
cumheights = F.pad(cumheights, pad=(1, 0), mode="constant", value=0.0)
|
139 |
+
cumheights = (top - bottom) * cumheights + bottom
|
140 |
+
cumheights[..., 0] = bottom
|
141 |
+
cumheights[..., -1] = top
|
142 |
+
heights = cumheights[..., 1:] - cumheights[..., :-1]
|
143 |
+
|
144 |
+
if inverse:
|
145 |
+
bin_idx = searchsorted(cumheights, inputs)[..., None]
|
146 |
+
else:
|
147 |
+
bin_idx = searchsorted(cumwidths, inputs)[..., None]
|
148 |
+
|
149 |
+
input_cumwidths = cumwidths.gather(-1, bin_idx)[..., 0]
|
150 |
+
input_bin_widths = widths.gather(-1, bin_idx)[..., 0]
|
151 |
+
|
152 |
+
input_cumheights = cumheights.gather(-1, bin_idx)[..., 0]
|
153 |
+
delta = heights / widths
|
154 |
+
input_delta = delta.gather(-1, bin_idx)[..., 0]
|
155 |
+
|
156 |
+
input_derivatives = derivatives.gather(-1, bin_idx)[..., 0]
|
157 |
+
input_derivatives_plus_one = derivatives[..., 1:].gather(-1, bin_idx)[..., 0]
|
158 |
+
|
159 |
+
input_heights = heights.gather(-1, bin_idx)[..., 0]
|
160 |
+
|
161 |
+
if inverse:
|
162 |
+
a = (inputs - input_cumheights) * (
|
163 |
+
input_derivatives + input_derivatives_plus_one - 2 * input_delta
|
164 |
+
) + input_heights * (input_delta - input_derivatives)
|
165 |
+
b = input_heights * input_derivatives - (inputs - input_cumheights) * (
|
166 |
+
input_derivatives + input_derivatives_plus_one - 2 * input_delta
|
167 |
+
)
|
168 |
+
c = -input_delta * (inputs - input_cumheights)
|
169 |
+
|
170 |
+
discriminant = b.pow(2) - 4 * a * c
|
171 |
+
assert (discriminant >= 0).all()
|
172 |
+
|
173 |
+
root = (2 * c) / (-b - torch.sqrt(discriminant))
|
174 |
+
outputs = root * input_bin_widths + input_cumwidths
|
175 |
+
|
176 |
+
theta_one_minus_theta = root * (1 - root)
|
177 |
+
denominator = input_delta + (
|
178 |
+
(input_derivatives + input_derivatives_plus_one - 2 * input_delta)
|
179 |
+
* theta_one_minus_theta
|
180 |
+
)
|
181 |
+
derivative_numerator = input_delta.pow(2) * (
|
182 |
+
input_derivatives_plus_one * root.pow(2)
|
183 |
+
+ 2 * input_delta * theta_one_minus_theta
|
184 |
+
+ input_derivatives * (1 - root).pow(2)
|
185 |
+
)
|
186 |
+
logabsdet = torch.log(derivative_numerator) - 2 * torch.log(denominator)
|
187 |
+
|
188 |
+
return outputs, -logabsdet
|
189 |
+
else:
|
190 |
+
theta = (inputs - input_cumwidths) / input_bin_widths
|
191 |
+
theta_one_minus_theta = theta * (1 - theta)
|
192 |
+
|
193 |
+
numerator = input_heights * (
|
194 |
+
input_delta * theta.pow(2) + input_derivatives * theta_one_minus_theta
|
195 |
+
)
|
196 |
+
denominator = input_delta + (
|
197 |
+
(input_derivatives + input_derivatives_plus_one - 2 * input_delta)
|
198 |
+
* theta_one_minus_theta
|
199 |
+
)
|
200 |
+
outputs = input_cumheights + numerator / denominator
|
201 |
+
|
202 |
+
derivative_numerator = input_delta.pow(2) * (
|
203 |
+
input_derivatives_plus_one * theta.pow(2)
|
204 |
+
+ 2 * input_delta * theta_one_minus_theta
|
205 |
+
+ input_derivatives * (1 - theta).pow(2)
|
206 |
+
)
|
207 |
+
logabsdet = torch.log(derivative_numerator) - 2 * torch.log(denominator)
|
208 |
+
|
209 |
+
return outputs, logabsdet
|
packages.txt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
build-essential
|
2 |
+
ffmpeg
|
3 |
+
aria2
|
requirements.txt
ADDED
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
gTTS
|
2 |
+
elevenlabs
|
3 |
+
stftpitchshift==1.5.1
|
4 |
+
torchcrepe
|
5 |
+
setuptools
|
6 |
+
wheel
|
7 |
+
httpx==0.23.0
|
8 |
+
faiss-gpu
|
9 |
+
fairseq
|
10 |
+
gradio==3.34.0
|
11 |
+
ffmpeg-python
|
12 |
+
praat-parselmouth
|
13 |
+
pyworld
|
14 |
+
numpy==1.23.5
|
15 |
+
i18n
|
16 |
+
numba==0.56.4
|
17 |
+
librosa==0.9.2
|
18 |
+
mega.py
|
19 |
+
gdown
|
20 |
+
onnxruntime
|
21 |
+
pyngrok==4.1.12
|
22 |
+
torch
|
rmvpe.py
ADDED
@@ -0,0 +1,432 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import sys, torch, numpy as np, traceback, pdb
|
2 |
+
import torch.nn as nn
|
3 |
+
from time import time as ttime
|
4 |
+
import torch.nn.functional as F
|
5 |
+
|
6 |
+
|
7 |
+
class BiGRU(nn.Module):
|
8 |
+
def __init__(self, input_features, hidden_features, num_layers):
|
9 |
+
super(BiGRU, self).__init__()
|
10 |
+
self.gru = nn.GRU(
|
11 |
+
input_features,
|
12 |
+
hidden_features,
|
13 |
+
num_layers=num_layers,
|
14 |
+
batch_first=True,
|
15 |
+
bidirectional=True,
|
16 |
+
)
|
17 |
+
|
18 |
+
def forward(self, x):
|
19 |
+
return self.gru(x)[0]
|
20 |
+
|
21 |
+
|
22 |
+
class ConvBlockRes(nn.Module):
|
23 |
+
def __init__(self, in_channels, out_channels, momentum=0.01):
|
24 |
+
super(ConvBlockRes, self).__init__()
|
25 |
+
self.conv = nn.Sequential(
|
26 |
+
nn.Conv2d(
|
27 |
+
in_channels=in_channels,
|
28 |
+
out_channels=out_channels,
|
29 |
+
kernel_size=(3, 3),
|
30 |
+
stride=(1, 1),
|
31 |
+
padding=(1, 1),
|
32 |
+
bias=False,
|
33 |
+
),
|
34 |
+
nn.BatchNorm2d(out_channels, momentum=momentum),
|
35 |
+
nn.ReLU(),
|
36 |
+
nn.Conv2d(
|
37 |
+
in_channels=out_channels,
|
38 |
+
out_channels=out_channels,
|
39 |
+
kernel_size=(3, 3),
|
40 |
+
stride=(1, 1),
|
41 |
+
padding=(1, 1),
|
42 |
+
bias=False,
|
43 |
+
),
|
44 |
+
nn.BatchNorm2d(out_channels, momentum=momentum),
|
45 |
+
nn.ReLU(),
|
46 |
+
)
|
47 |
+
if in_channels != out_channels:
|
48 |
+
self.shortcut = nn.Conv2d(in_channels, out_channels, (1, 1))
|
49 |
+
self.is_shortcut = True
|
50 |
+
else:
|
51 |
+
self.is_shortcut = False
|
52 |
+
|
53 |
+
def forward(self, x):
|
54 |
+
if self.is_shortcut:
|
55 |
+
return self.conv(x) + self.shortcut(x)
|
56 |
+
else:
|
57 |
+
return self.conv(x) + x
|
58 |
+
|
59 |
+
|
60 |
+
class Encoder(nn.Module):
|
61 |
+
def __init__(
|
62 |
+
self,
|
63 |
+
in_channels,
|
64 |
+
in_size,
|
65 |
+
n_encoders,
|
66 |
+
kernel_size,
|
67 |
+
n_blocks,
|
68 |
+
out_channels=16,
|
69 |
+
momentum=0.01,
|
70 |
+
):
|
71 |
+
super(Encoder, self).__init__()
|
72 |
+
self.n_encoders = n_encoders
|
73 |
+
self.bn = nn.BatchNorm2d(in_channels, momentum=momentum)
|
74 |
+
self.layers = nn.ModuleList()
|
75 |
+
self.latent_channels = []
|
76 |
+
for i in range(self.n_encoders):
|
77 |
+
self.layers.append(
|
78 |
+
ResEncoderBlock(
|
79 |
+
in_channels, out_channels, kernel_size, n_blocks, momentum=momentum
|
80 |
+
)
|
81 |
+
)
|
82 |
+
self.latent_channels.append([out_channels, in_size])
|
83 |
+
in_channels = out_channels
|
84 |
+
out_channels *= 2
|
85 |
+
in_size //= 2
|
86 |
+
self.out_size = in_size
|
87 |
+
self.out_channel = out_channels
|
88 |
+
|
89 |
+
def forward(self, x):
|
90 |
+
concat_tensors = []
|
91 |
+
x = self.bn(x)
|
92 |
+
for i in range(self.n_encoders):
|
93 |
+
_, x = self.layers[i](x)
|
94 |
+
concat_tensors.append(_)
|
95 |
+
return x, concat_tensors
|
96 |
+
|
97 |
+
|
98 |
+
class ResEncoderBlock(nn.Module):
|
99 |
+
def __init__(
|
100 |
+
self, in_channels, out_channels, kernel_size, n_blocks=1, momentum=0.01
|
101 |
+
):
|
102 |
+
super(ResEncoderBlock, self).__init__()
|
103 |
+
self.n_blocks = n_blocks
|
104 |
+
self.conv = nn.ModuleList()
|
105 |
+
self.conv.append(ConvBlockRes(in_channels, out_channels, momentum))
|
106 |
+
for i in range(n_blocks - 1):
|
107 |
+
self.conv.append(ConvBlockRes(out_channels, out_channels, momentum))
|
108 |
+
self.kernel_size = kernel_size
|
109 |
+
if self.kernel_size is not None:
|
110 |
+
self.pool = nn.AvgPool2d(kernel_size=kernel_size)
|
111 |
+
|
112 |
+
def forward(self, x):
|
113 |
+
for i in range(self.n_blocks):
|
114 |
+
x = self.conv[i](x)
|
115 |
+
if self.kernel_size is not None:
|
116 |
+
return x, self.pool(x)
|
117 |
+
else:
|
118 |
+
return x
|
119 |
+
|
120 |
+
|
121 |
+
class Intermediate(nn.Module): #
|
122 |
+
def __init__(self, in_channels, out_channels, n_inters, n_blocks, momentum=0.01):
|
123 |
+
super(Intermediate, self).__init__()
|
124 |
+
self.n_inters = n_inters
|
125 |
+
self.layers = nn.ModuleList()
|
126 |
+
self.layers.append(
|
127 |
+
ResEncoderBlock(in_channels, out_channels, None, n_blocks, momentum)
|
128 |
+
)
|
129 |
+
for i in range(self.n_inters - 1):
|
130 |
+
self.layers.append(
|
131 |
+
ResEncoderBlock(out_channels, out_channels, None, n_blocks, momentum)
|
132 |
+
)
|
133 |
+
|
134 |
+
def forward(self, x):
|
135 |
+
for i in range(self.n_inters):
|
136 |
+
x = self.layers[i](x)
|
137 |
+
return x
|
138 |
+
|
139 |
+
|
140 |
+
class ResDecoderBlock(nn.Module):
|
141 |
+
def __init__(self, in_channels, out_channels, stride, n_blocks=1, momentum=0.01):
|
142 |
+
super(ResDecoderBlock, self).__init__()
|
143 |
+
out_padding = (0, 1) if stride == (1, 2) else (1, 1)
|
144 |
+
self.n_blocks = n_blocks
|
145 |
+
self.conv1 = nn.Sequential(
|
146 |
+
nn.ConvTranspose2d(
|
147 |
+
in_channels=in_channels,
|
148 |
+
out_channels=out_channels,
|
149 |
+
kernel_size=(3, 3),
|
150 |
+
stride=stride,
|
151 |
+
padding=(1, 1),
|
152 |
+
output_padding=out_padding,
|
153 |
+
bias=False,
|
154 |
+
),
|
155 |
+
nn.BatchNorm2d(out_channels, momentum=momentum),
|
156 |
+
nn.ReLU(),
|
157 |
+
)
|
158 |
+
self.conv2 = nn.ModuleList()
|
159 |
+
self.conv2.append(ConvBlockRes(out_channels * 2, out_channels, momentum))
|
160 |
+
for i in range(n_blocks - 1):
|
161 |
+
self.conv2.append(ConvBlockRes(out_channels, out_channels, momentum))
|
162 |
+
|
163 |
+
def forward(self, x, concat_tensor):
|
164 |
+
x = self.conv1(x)
|
165 |
+
x = torch.cat((x, concat_tensor), dim=1)
|
166 |
+
for i in range(self.n_blocks):
|
167 |
+
x = self.conv2[i](x)
|
168 |
+
return x
|
169 |
+
|
170 |
+
|
171 |
+
class Decoder(nn.Module):
|
172 |
+
def __init__(self, in_channels, n_decoders, stride, n_blocks, momentum=0.01):
|
173 |
+
super(Decoder, self).__init__()
|
174 |
+
self.layers = nn.ModuleList()
|
175 |
+
self.n_decoders = n_decoders
|
176 |
+
for i in range(self.n_decoders):
|
177 |
+
out_channels = in_channels // 2
|
178 |
+
self.layers.append(
|
179 |
+
ResDecoderBlock(in_channels, out_channels, stride, n_blocks, momentum)
|
180 |
+
)
|
181 |
+
in_channels = out_channels
|
182 |
+
|
183 |
+
def forward(self, x, concat_tensors):
|
184 |
+
for i in range(self.n_decoders):
|
185 |
+
x = self.layers[i](x, concat_tensors[-1 - i])
|
186 |
+
return x
|
187 |
+
|
188 |
+
|
189 |
+
class DeepUnet(nn.Module):
|
190 |
+
def __init__(
|
191 |
+
self,
|
192 |
+
kernel_size,
|
193 |
+
n_blocks,
|
194 |
+
en_de_layers=5,
|
195 |
+
inter_layers=4,
|
196 |
+
in_channels=1,
|
197 |
+
en_out_channels=16,
|
198 |
+
):
|
199 |
+
super(DeepUnet, self).__init__()
|
200 |
+
self.encoder = Encoder(
|
201 |
+
in_channels, 128, en_de_layers, kernel_size, n_blocks, en_out_channels
|
202 |
+
)
|
203 |
+
self.intermediate = Intermediate(
|
204 |
+
self.encoder.out_channel // 2,
|
205 |
+
self.encoder.out_channel,
|
206 |
+
inter_layers,
|
207 |
+
n_blocks,
|
208 |
+
)
|
209 |
+
self.decoder = Decoder(
|
210 |
+
self.encoder.out_channel, en_de_layers, kernel_size, n_blocks
|
211 |
+
)
|
212 |
+
|
213 |
+
def forward(self, x):
|
214 |
+
x, concat_tensors = self.encoder(x)
|
215 |
+
x = self.intermediate(x)
|
216 |
+
x = self.decoder(x, concat_tensors)
|
217 |
+
return x
|
218 |
+
|
219 |
+
|
220 |
+
class E2E(nn.Module):
|
221 |
+
def __init__(
|
222 |
+
self,
|
223 |
+
n_blocks,
|
224 |
+
n_gru,
|
225 |
+
kernel_size,
|
226 |
+
en_de_layers=5,
|
227 |
+
inter_layers=4,
|
228 |
+
in_channels=1,
|
229 |
+
en_out_channels=16,
|
230 |
+
):
|
231 |
+
super(E2E, self).__init__()
|
232 |
+
self.unet = DeepUnet(
|
233 |
+
kernel_size,
|
234 |
+
n_blocks,
|
235 |
+
en_de_layers,
|
236 |
+
inter_layers,
|
237 |
+
in_channels,
|
238 |
+
en_out_channels,
|
239 |
+
)
|
240 |
+
self.cnn = nn.Conv2d(en_out_channels, 3, (3, 3), padding=(1, 1))
|
241 |
+
if n_gru:
|
242 |
+
self.fc = nn.Sequential(
|
243 |
+
BiGRU(3 * 128, 256, n_gru),
|
244 |
+
nn.Linear(512, 360),
|
245 |
+
nn.Dropout(0.25),
|
246 |
+
nn.Sigmoid(),
|
247 |
+
)
|
248 |
+
else:
|
249 |
+
self.fc = nn.Sequential(
|
250 |
+
nn.Linear(3 * N_MELS, N_CLASS), nn.Dropout(0.25), nn.Sigmoid()
|
251 |
+
)
|
252 |
+
|
253 |
+
def forward(self, mel):
|
254 |
+
mel = mel.transpose(-1, -2).unsqueeze(1)
|
255 |
+
x = self.cnn(self.unet(mel)).transpose(1, 2).flatten(-2)
|
256 |
+
x = self.fc(x)
|
257 |
+
return x
|
258 |
+
|
259 |
+
|
260 |
+
from librosa.filters import mel
|
261 |
+
|
262 |
+
|
263 |
+
class MelSpectrogram(torch.nn.Module):
|
264 |
+
def __init__(
|
265 |
+
self,
|
266 |
+
is_half,
|
267 |
+
n_mel_channels,
|
268 |
+
sampling_rate,
|
269 |
+
win_length,
|
270 |
+
hop_length,
|
271 |
+
n_fft=None,
|
272 |
+
mel_fmin=0,
|
273 |
+
mel_fmax=None,
|
274 |
+
clamp=1e-5,
|
275 |
+
):
|
276 |
+
super().__init__()
|
277 |
+
n_fft = win_length if n_fft is None else n_fft
|
278 |
+
self.hann_window = {}
|
279 |
+
mel_basis = mel(
|
280 |
+
sr=sampling_rate,
|
281 |
+
n_fft=n_fft,
|
282 |
+
n_mels=n_mel_channels,
|
283 |
+
fmin=mel_fmin,
|
284 |
+
fmax=mel_fmax,
|
285 |
+
htk=True,
|
286 |
+
)
|
287 |
+
mel_basis = torch.from_numpy(mel_basis).float()
|
288 |
+
self.register_buffer("mel_basis", mel_basis)
|
289 |
+
self.n_fft = win_length if n_fft is None else n_fft
|
290 |
+
self.hop_length = hop_length
|
291 |
+
self.win_length = win_length
|
292 |
+
self.sampling_rate = sampling_rate
|
293 |
+
self.n_mel_channels = n_mel_channels
|
294 |
+
self.clamp = clamp
|
295 |
+
self.is_half = is_half
|
296 |
+
|
297 |
+
def forward(self, audio, keyshift=0, speed=1, center=True):
|
298 |
+
factor = 2 ** (keyshift / 12)
|
299 |
+
n_fft_new = int(np.round(self.n_fft * factor))
|
300 |
+
win_length_new = int(np.round(self.win_length * factor))
|
301 |
+
hop_length_new = int(np.round(self.hop_length * speed))
|
302 |
+
keyshift_key = str(keyshift) + "_" + str(audio.device)
|
303 |
+
if keyshift_key not in self.hann_window:
|
304 |
+
self.hann_window[keyshift_key] = torch.hann_window(win_length_new).to(
|
305 |
+
audio.device
|
306 |
+
)
|
307 |
+
fft = torch.stft(
|
308 |
+
audio,
|
309 |
+
n_fft=n_fft_new,
|
310 |
+
hop_length=hop_length_new,
|
311 |
+
win_length=win_length_new,
|
312 |
+
window=self.hann_window[keyshift_key],
|
313 |
+
center=center,
|
314 |
+
return_complex=True,
|
315 |
+
)
|
316 |
+
magnitude = torch.sqrt(fft.real.pow(2) + fft.imag.pow(2))
|
317 |
+
if keyshift != 0:
|
318 |
+
size = self.n_fft // 2 + 1
|
319 |
+
resize = magnitude.size(1)
|
320 |
+
if resize < size:
|
321 |
+
magnitude = F.pad(magnitude, (0, 0, 0, size - resize))
|
322 |
+
magnitude = magnitude[:, :size, :] * self.win_length / win_length_new
|
323 |
+
mel_output = torch.matmul(self.mel_basis, magnitude)
|
324 |
+
if self.is_half == True:
|
325 |
+
mel_output = mel_output.half()
|
326 |
+
log_mel_spec = torch.log(torch.clamp(mel_output, min=self.clamp))
|
327 |
+
return log_mel_spec
|
328 |
+
|
329 |
+
|
330 |
+
class RMVPE:
|
331 |
+
def __init__(self, model_path, is_half, device=None):
|
332 |
+
self.resample_kernel = {}
|
333 |
+
model = E2E(4, 1, (2, 2))
|
334 |
+
ckpt = torch.load(model_path, map_location="cpu")
|
335 |
+
model.load_state_dict(ckpt)
|
336 |
+
model.eval()
|
337 |
+
if is_half == True:
|
338 |
+
model = model.half()
|
339 |
+
self.model = model
|
340 |
+
self.resample_kernel = {}
|
341 |
+
self.is_half = is_half
|
342 |
+
if device is None:
|
343 |
+
device = "cuda" if torch.cuda.is_available() else "cpu"
|
344 |
+
self.device = device
|
345 |
+
self.mel_extractor = MelSpectrogram(
|
346 |
+
is_half, 128, 16000, 1024, 160, None, 30, 8000
|
347 |
+
).to(device)
|
348 |
+
self.model = self.model.to(device)
|
349 |
+
cents_mapping = 20 * np.arange(360) + 1997.3794084376191
|
350 |
+
self.cents_mapping = np.pad(cents_mapping, (4, 4)) # 368
|
351 |
+
|
352 |
+
def mel2hidden(self, mel):
|
353 |
+
with torch.no_grad():
|
354 |
+
n_frames = mel.shape[-1]
|
355 |
+
mel = F.pad(
|
356 |
+
mel, (0, 32 * ((n_frames - 1) // 32 + 1) - n_frames), mode="reflect"
|
357 |
+
)
|
358 |
+
hidden = self.model(mel)
|
359 |
+
return hidden[:, :n_frames]
|
360 |
+
|
361 |
+
def decode(self, hidden, thred=0.03):
|
362 |
+
cents_pred = self.to_local_average_cents(hidden, thred=thred)
|
363 |
+
f0 = 10 * (2 ** (cents_pred / 1200))
|
364 |
+
f0[f0 == 10] = 0
|
365 |
+
# f0 = np.array([10 * (2 ** (cent_pred / 1200)) if cent_pred else 0 for cent_pred in cents_pred])
|
366 |
+
return f0
|
367 |
+
|
368 |
+
def infer_from_audio(self, audio, thred=0.03):
|
369 |
+
audio = torch.from_numpy(audio).float().to(self.device).unsqueeze(0)
|
370 |
+
# torch.cuda.synchronize()
|
371 |
+
# t0=ttime()
|
372 |
+
mel = self.mel_extractor(audio, center=True)
|
373 |
+
# torch.cuda.synchronize()
|
374 |
+
# t1=ttime()
|
375 |
+
hidden = self.mel2hidden(mel)
|
376 |
+
# torch.cuda.synchronize()
|
377 |
+
# t2=ttime()
|
378 |
+
hidden = hidden.squeeze(0).cpu().numpy()
|
379 |
+
if self.is_half == True:
|
380 |
+
hidden = hidden.astype("float32")
|
381 |
+
f0 = self.decode(hidden, thred=thred)
|
382 |
+
# torch.cuda.synchronize()
|
383 |
+
# t3=ttime()
|
384 |
+
# print("hmvpe:%s\t%s\t%s\t%s"%(t1-t0,t2-t1,t3-t2,t3-t0))
|
385 |
+
return f0
|
386 |
+
|
387 |
+
def to_local_average_cents(self, salience, thred=0.05):
|
388 |
+
# t0 = ttime()
|
389 |
+
center = np.argmax(salience, axis=1) # 帧长#index
|
390 |
+
salience = np.pad(salience, ((0, 0), (4, 4))) # 帧长,368
|
391 |
+
# t1 = ttime()
|
392 |
+
center += 4
|
393 |
+
todo_salience = []
|
394 |
+
todo_cents_mapping = []
|
395 |
+
starts = center - 4
|
396 |
+
ends = center + 5
|
397 |
+
for idx in range(salience.shape[0]):
|
398 |
+
todo_salience.append(salience[:, starts[idx] : ends[idx]][idx])
|
399 |
+
todo_cents_mapping.append(self.cents_mapping[starts[idx] : ends[idx]])
|
400 |
+
# t2 = ttime()
|
401 |
+
todo_salience = np.array(todo_salience) # 帧长,9
|
402 |
+
todo_cents_mapping = np.array(todo_cents_mapping) # 帧长,9
|
403 |
+
product_sum = np.sum(todo_salience * todo_cents_mapping, 1)
|
404 |
+
weight_sum = np.sum(todo_salience, 1) # 帧长
|
405 |
+
devided = product_sum / weight_sum # 帧长
|
406 |
+
# t3 = ttime()
|
407 |
+
maxx = np.max(salience, axis=1) # 帧长
|
408 |
+
devided[maxx <= thred] = 0
|
409 |
+
# t4 = ttime()
|
410 |
+
# print("decode:%s\t%s\t%s\t%s" % (t1 - t0, t2 - t1, t3 - t2, t4 - t3))
|
411 |
+
return devided
|
412 |
+
|
413 |
+
|
414 |
+
# if __name__ == '__main__':
|
415 |
+
# audio, sampling_rate = sf.read("卢本伟语录~1.wav")
|
416 |
+
# if len(audio.shape) > 1:
|
417 |
+
# audio = librosa.to_mono(audio.transpose(1, 0))
|
418 |
+
# audio_bak = audio.copy()
|
419 |
+
# if sampling_rate != 16000:
|
420 |
+
# audio = librosa.resample(audio, orig_sr=sampling_rate, target_sr=16000)
|
421 |
+
# model_path = "/bili-coeus/jupyter/jupyterhub-liujing04/vits_ch/test-RMVPE/weights/rmvpe_llc_half.pt"
|
422 |
+
# thred = 0.03 # 0.01
|
423 |
+
# device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
424 |
+
# rmvpe = RMVPE(model_path,is_half=False, device=device)
|
425 |
+
# t0=ttime()
|
426 |
+
# f0 = rmvpe.infer_from_audio(audio, thred=thred)
|
427 |
+
# f0 = rmvpe.infer_from_audio(audio, thred=thred)
|
428 |
+
# f0 = rmvpe.infer_from_audio(audio, thred=thred)
|
429 |
+
# f0 = rmvpe.infer_from_audio(audio, thred=thred)
|
430 |
+
# f0 = rmvpe.infer_from_audio(audio, thred=thred)
|
431 |
+
# t1=ttime()
|
432 |
+
# print(f0.shape,t1-t0)
|
run.sh
ADDED
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Install Debian packages
|
2 |
+
sudo apt-get update
|
3 |
+
sudo apt-get install -qq -y build-essential ffmpeg aria2
|
4 |
+
|
5 |
+
# Upgrade pip and setuptools
|
6 |
+
pip install --upgrade pip
|
7 |
+
pip install --upgrade setuptools
|
8 |
+
|
9 |
+
# Install wheel package (built-package format for Python)
|
10 |
+
pip install wheel
|
11 |
+
|
12 |
+
# Install Python packages using pip
|
13 |
+
pip install -r requirements.txt
|
14 |
+
|
15 |
+
# Run application locally at http://127.0.0.1:7860
|
16 |
+
python app.py
|
utils.py
ADDED
@@ -0,0 +1,151 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import ffmpeg
|
2 |
+
import numpy as np
|
3 |
+
|
4 |
+
# import praatio
|
5 |
+
# import praatio.praat_scripts
|
6 |
+
import os
|
7 |
+
import sys
|
8 |
+
|
9 |
+
import random
|
10 |
+
|
11 |
+
import csv
|
12 |
+
|
13 |
+
platform_stft_mapping = {
|
14 |
+
"linux": "stftpitchshift",
|
15 |
+
"darwin": "stftpitchshift",
|
16 |
+
"win32": "stftpitchshift.exe",
|
17 |
+
}
|
18 |
+
|
19 |
+
stft = platform_stft_mapping.get(sys.platform)
|
20 |
+
# praatEXE = join('.',os.path.abspath(os.getcwd()) + r"\Praat.exe")
|
21 |
+
|
22 |
+
|
23 |
+
def CSVutil(file, rw, type, *args):
|
24 |
+
if type == "formanting":
|
25 |
+
if rw == "r":
|
26 |
+
with open(file) as fileCSVread:
|
27 |
+
csv_reader = list(csv.reader(fileCSVread))
|
28 |
+
return (
|
29 |
+
(csv_reader[0][0], csv_reader[0][1], csv_reader[0][2])
|
30 |
+
if csv_reader is not None
|
31 |
+
else (lambda: exec('raise ValueError("No data")'))()
|
32 |
+
)
|
33 |
+
else:
|
34 |
+
if args:
|
35 |
+
doformnt = args[0]
|
36 |
+
else:
|
37 |
+
doformnt = False
|
38 |
+
qfr = args[1] if len(args) > 1 else 1.0
|
39 |
+
tmb = args[2] if len(args) > 2 else 1.0
|
40 |
+
with open(file, rw, newline="") as fileCSVwrite:
|
41 |
+
csv_writer = csv.writer(fileCSVwrite, delimiter=",")
|
42 |
+
csv_writer.writerow([doformnt, qfr, tmb])
|
43 |
+
elif type == "stop":
|
44 |
+
stop = args[0] if args else False
|
45 |
+
with open(file, rw, newline="") as fileCSVwrite:
|
46 |
+
csv_writer = csv.writer(fileCSVwrite, delimiter=",")
|
47 |
+
csv_writer.writerow([stop])
|
48 |
+
|
49 |
+
|
50 |
+
def load_audio(file, sr, DoFormant, Quefrency, Timbre):
|
51 |
+
converted = False
|
52 |
+
DoFormant, Quefrency, Timbre = CSVutil("csvdb/formanting.csv", "r", "formanting")
|
53 |
+
try:
|
54 |
+
# https://github.com/openai/whisper/blob/main/whisper/audio.py#L26
|
55 |
+
# This launches a subprocess to decode audio while down-mixing and resampling as necessary.
|
56 |
+
# Requires the ffmpeg CLI and `ffmpeg-python` package to be installed.
|
57 |
+
file = (
|
58 |
+
file.strip(" ").strip('"').strip("\n").strip('"').strip(" ")
|
59 |
+
) # 防止小白拷路径头尾带了空格和"和回车
|
60 |
+
file_formanted = file.strip(" ").strip('"').strip("\n").strip('"').strip(" ")
|
61 |
+
|
62 |
+
# print(f"dofor={bool(DoFormant)} timbr={Timbre} quef={Quefrency}\n")
|
63 |
+
|
64 |
+
if (
|
65 |
+
lambda DoFormant: True
|
66 |
+
if DoFormant.lower() == "true"
|
67 |
+
else (False if DoFormant.lower() == "false" else DoFormant)
|
68 |
+
)(DoFormant):
|
69 |
+
numerator = round(random.uniform(1, 4), 4)
|
70 |
+
# os.system(f"stftpitchshift -i {file} -q {Quefrency} -t {Timbre} -o {file_formanted}")
|
71 |
+
# print('stftpitchshift -i "%s" -p 1.0 --rms -w 128 -v 8 -q %s -t %s -o "%s"' % (file, Quefrency, Timbre, file_formanted))
|
72 |
+
|
73 |
+
if not file.endswith(".wav"):
|
74 |
+
if not os.path.isfile(f"{file_formanted}.wav"):
|
75 |
+
converted = True
|
76 |
+
# print(f"\nfile = {file}\n")
|
77 |
+
# print(f"\nfile_formanted = {file_formanted}\n")
|
78 |
+
converting = (
|
79 |
+
ffmpeg.input(file_formanted, threads=0)
|
80 |
+
.output(f"{file_formanted}.wav")
|
81 |
+
.run(
|
82 |
+
cmd=["ffmpeg", "-nostdin"],
|
83 |
+
capture_stdout=True,
|
84 |
+
capture_stderr=True,
|
85 |
+
)
|
86 |
+
)
|
87 |
+
else:
|
88 |
+
pass
|
89 |
+
|
90 |
+
file_formanted = (
|
91 |
+
f"{file_formanted}.wav"
|
92 |
+
if not file_formanted.endswith(".wav")
|
93 |
+
else file_formanted
|
94 |
+
)
|
95 |
+
|
96 |
+
print(f" · Formanting {file_formanted}...\n")
|
97 |
+
|
98 |
+
os.system(
|
99 |
+
'%s -i "%s" -q "%s" -t "%s" -o "%sFORMANTED_%s.wav"'
|
100 |
+
% (
|
101 |
+
stft,
|
102 |
+
file_formanted,
|
103 |
+
Quefrency,
|
104 |
+
Timbre,
|
105 |
+
file_formanted,
|
106 |
+
str(numerator),
|
107 |
+
)
|
108 |
+
)
|
109 |
+
|
110 |
+
print(f" · Formanted {file_formanted}!\n")
|
111 |
+
|
112 |
+
# filepraat = (os.path.abspath(os.getcwd()) + '\\' + file).replace('/','\\')
|
113 |
+
# file_formantedpraat = ('"' + os.path.abspath(os.getcwd()) + '/' + 'formanted'.join(file_formanted) + '"').replace('/','\\')
|
114 |
+
# print("%sFORMANTED_%s.wav" % (file_formanted, str(numerator)))
|
115 |
+
|
116 |
+
out, _ = (
|
117 |
+
ffmpeg.input(
|
118 |
+
"%sFORMANTED_%s.wav" % (file_formanted, str(numerator)), threads=0
|
119 |
+
)
|
120 |
+
.output("-", format="f32le", acodec="pcm_f32le", ac=1, ar=sr)
|
121 |
+
.run(
|
122 |
+
cmd=["ffmpeg", "-nostdin"], capture_stdout=True, capture_stderr=True
|
123 |
+
)
|
124 |
+
)
|
125 |
+
|
126 |
+
try:
|
127 |
+
os.remove("%sFORMANTED_%s.wav" % (file_formanted, str(numerator)))
|
128 |
+
except Exception:
|
129 |
+
pass
|
130 |
+
print("couldn't remove formanted type of file")
|
131 |
+
|
132 |
+
else:
|
133 |
+
out, _ = (
|
134 |
+
ffmpeg.input(file, threads=0)
|
135 |
+
.output("-", format="f32le", acodec="pcm_f32le", ac=1, ar=sr)
|
136 |
+
.run(
|
137 |
+
cmd=["ffmpeg", "-nostdin"], capture_stdout=True, capture_stderr=True
|
138 |
+
)
|
139 |
+
)
|
140 |
+
except Exception as e:
|
141 |
+
raise RuntimeError(f"Failed to load audio: {e}")
|
142 |
+
|
143 |
+
if converted:
|
144 |
+
try:
|
145 |
+
os.remove(file_formanted)
|
146 |
+
except Exception:
|
147 |
+
pass
|
148 |
+
print("couldn't remove converted type of file")
|
149 |
+
converted = False
|
150 |
+
|
151 |
+
return np.frombuffer(out, np.float32).flatten()
|
vc_infer_pipeline.py
ADDED
@@ -0,0 +1,646 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import numpy as np, parselmouth, torch, pdb, sys, os
|
2 |
+
from time import time as ttime
|
3 |
+
import torch.nn.functional as F
|
4 |
+
import torchcrepe # Fork feature. Use the crepe f0 algorithm. New dependency (pip install torchcrepe)
|
5 |
+
from torch import Tensor
|
6 |
+
import scipy.signal as signal
|
7 |
+
import pyworld, os, traceback, faiss, librosa, torchcrepe
|
8 |
+
from scipy import signal
|
9 |
+
from functools import lru_cache
|
10 |
+
|
11 |
+
now_dir = os.getcwd()
|
12 |
+
sys.path.append(now_dir)
|
13 |
+
|
14 |
+
bh, ah = signal.butter(N=5, Wn=48, btype="high", fs=16000)
|
15 |
+
|
16 |
+
input_audio_path2wav = {}
|
17 |
+
|
18 |
+
|
19 |
+
@lru_cache
|
20 |
+
def cache_harvest_f0(input_audio_path, fs, f0max, f0min, frame_period):
|
21 |
+
audio = input_audio_path2wav[input_audio_path]
|
22 |
+
f0, t = pyworld.harvest(
|
23 |
+
audio,
|
24 |
+
fs=fs,
|
25 |
+
f0_ceil=f0max,
|
26 |
+
f0_floor=f0min,
|
27 |
+
frame_period=frame_period,
|
28 |
+
)
|
29 |
+
f0 = pyworld.stonemask(audio, f0, t, fs)
|
30 |
+
return f0
|
31 |
+
|
32 |
+
|
33 |
+
def change_rms(data1, sr1, data2, sr2, rate): # 1是输入音频,2是输出音频,rate是2的占比
|
34 |
+
# print(data1.max(),data2.max())
|
35 |
+
rms1 = librosa.feature.rms(
|
36 |
+
y=data1, frame_length=sr1 // 2 * 2, hop_length=sr1 // 2
|
37 |
+
) # 每半秒一个点
|
38 |
+
rms2 = librosa.feature.rms(y=data2, frame_length=sr2 // 2 * 2, hop_length=sr2 // 2)
|
39 |
+
rms1 = torch.from_numpy(rms1)
|
40 |
+
rms1 = F.interpolate(
|
41 |
+
rms1.unsqueeze(0), size=data2.shape[0], mode="linear"
|
42 |
+
).squeeze()
|
43 |
+
rms2 = torch.from_numpy(rms2)
|
44 |
+
rms2 = F.interpolate(
|
45 |
+
rms2.unsqueeze(0), size=data2.shape[0], mode="linear"
|
46 |
+
).squeeze()
|
47 |
+
rms2 = torch.max(rms2, torch.zeros_like(rms2) + 1e-6)
|
48 |
+
data2 *= (
|
49 |
+
torch.pow(rms1, torch.tensor(1 - rate))
|
50 |
+
* torch.pow(rms2, torch.tensor(rate - 1))
|
51 |
+
).numpy()
|
52 |
+
return data2
|
53 |
+
|
54 |
+
|
55 |
+
class VC(object):
|
56 |
+
def __init__(self, tgt_sr, config):
|
57 |
+
self.x_pad, self.x_query, self.x_center, self.x_max, self.is_half = (
|
58 |
+
config.x_pad,
|
59 |
+
config.x_query,
|
60 |
+
config.x_center,
|
61 |
+
config.x_max,
|
62 |
+
config.is_half,
|
63 |
+
)
|
64 |
+
self.sr = 16000 # hubert输入采样率
|
65 |
+
self.window = 160 # 每帧点数
|
66 |
+
self.t_pad = self.sr * self.x_pad # 每条前后pad时间
|
67 |
+
self.t_pad_tgt = tgt_sr * self.x_pad
|
68 |
+
self.t_pad2 = self.t_pad * 2
|
69 |
+
self.t_query = self.sr * self.x_query # 查询切点前后查询时间
|
70 |
+
self.t_center = self.sr * self.x_center # 查询切点位置
|
71 |
+
self.t_max = self.sr * self.x_max # 免查询时长阈值
|
72 |
+
self.device = config.device
|
73 |
+
|
74 |
+
# Fork Feature: Get the best torch device to use for f0 algorithms that require a torch device. Will return the type (torch.device)
|
75 |
+
def get_optimal_torch_device(self, index: int = 0) -> torch.device:
|
76 |
+
# Get cuda device
|
77 |
+
if torch.cuda.is_available():
|
78 |
+
return torch.device(
|
79 |
+
f"cuda:{index % torch.cuda.device_count()}"
|
80 |
+
) # Very fast
|
81 |
+
elif torch.backends.mps.is_available():
|
82 |
+
return torch.device("mps")
|
83 |
+
# Insert an else here to grab "xla" devices if available. TO DO later. Requires the torch_xla.core.xla_model library
|
84 |
+
# Else wise return the "cpu" as a torch device,
|
85 |
+
return torch.device("cpu")
|
86 |
+
|
87 |
+
# Fork Feature: Compute f0 with the crepe method
|
88 |
+
def get_f0_crepe_computation(
|
89 |
+
self,
|
90 |
+
x,
|
91 |
+
f0_min,
|
92 |
+
f0_max,
|
93 |
+
p_len,
|
94 |
+
hop_length=160, # 512 before. Hop length changes the speed that the voice jumps to a different dramatic pitch. Lower hop lengths means more pitch accuracy but longer inference time.
|
95 |
+
model="full", # Either use crepe-tiny "tiny" or crepe "full". Default is full
|
96 |
+
):
|
97 |
+
x = x.astype(
|
98 |
+
np.float32
|
99 |
+
) # fixes the F.conv2D exception. We needed to convert double to float.
|
100 |
+
x /= np.quantile(np.abs(x), 0.999)
|
101 |
+
torch_device = self.get_optimal_torch_device()
|
102 |
+
audio = torch.from_numpy(x).to(torch_device, copy=True)
|
103 |
+
audio = torch.unsqueeze(audio, dim=0)
|
104 |
+
if audio.ndim == 2 and audio.shape[0] > 1:
|
105 |
+
audio = torch.mean(audio, dim=0, keepdim=True).detach()
|
106 |
+
audio = audio.detach()
|
107 |
+
print("Initiating prediction with a crepe_hop_length of: " + str(hop_length))
|
108 |
+
pitch: Tensor = torchcrepe.predict(
|
109 |
+
audio,
|
110 |
+
self.sr,
|
111 |
+
hop_length,
|
112 |
+
f0_min,
|
113 |
+
f0_max,
|
114 |
+
model,
|
115 |
+
batch_size=hop_length * 2,
|
116 |
+
device=torch_device,
|
117 |
+
pad=True,
|
118 |
+
)
|
119 |
+
p_len = p_len or x.shape[0] // hop_length
|
120 |
+
# Resize the pitch for final f0
|
121 |
+
source = np.array(pitch.squeeze(0).cpu().float().numpy())
|
122 |
+
source[source < 0.001] = np.nan
|
123 |
+
target = np.interp(
|
124 |
+
np.arange(0, len(source) * p_len, len(source)) / p_len,
|
125 |
+
np.arange(0, len(source)),
|
126 |
+
source,
|
127 |
+
)
|
128 |
+
f0 = np.nan_to_num(target)
|
129 |
+
return f0 # Resized f0
|
130 |
+
|
131 |
+
def get_f0_official_crepe_computation(
|
132 |
+
self,
|
133 |
+
x,
|
134 |
+
f0_min,
|
135 |
+
f0_max,
|
136 |
+
model="full",
|
137 |
+
):
|
138 |
+
# Pick a batch size that doesn't cause memory errors on your gpu
|
139 |
+
batch_size = 512
|
140 |
+
# Compute pitch using first gpu
|
141 |
+
audio = torch.tensor(np.copy(x))[None].float()
|
142 |
+
f0, pd = torchcrepe.predict(
|
143 |
+
audio,
|
144 |
+
self.sr,
|
145 |
+
self.window,
|
146 |
+
f0_min,
|
147 |
+
f0_max,
|
148 |
+
model,
|
149 |
+
batch_size=batch_size,
|
150 |
+
device=self.device,
|
151 |
+
return_periodicity=True,
|
152 |
+
)
|
153 |
+
pd = torchcrepe.filter.median(pd, 3)
|
154 |
+
f0 = torchcrepe.filter.mean(f0, 3)
|
155 |
+
f0[pd < 0.1] = 0
|
156 |
+
f0 = f0[0].cpu().numpy()
|
157 |
+
return f0
|
158 |
+
|
159 |
+
# Fork Feature: Compute pYIN f0 method
|
160 |
+
def get_f0_pyin_computation(self, x, f0_min, f0_max):
|
161 |
+
y, sr = librosa.load("saudio/Sidney.wav", self.sr, mono=True)
|
162 |
+
f0, _, _ = librosa.pyin(y, sr=self.sr, fmin=f0_min, fmax=f0_max)
|
163 |
+
f0 = f0[1:] # Get rid of extra first frame
|
164 |
+
return f0
|
165 |
+
|
166 |
+
# Fork Feature: Acquire median hybrid f0 estimation calculation
|
167 |
+
def get_f0_hybrid_computation(
|
168 |
+
self,
|
169 |
+
methods_str,
|
170 |
+
input_audio_path,
|
171 |
+
x,
|
172 |
+
f0_min,
|
173 |
+
f0_max,
|
174 |
+
p_len,
|
175 |
+
filter_radius,
|
176 |
+
crepe_hop_length,
|
177 |
+
time_step,
|
178 |
+
):
|
179 |
+
# Get various f0 methods from input to use in the computation stack
|
180 |
+
s = methods_str
|
181 |
+
s = s.split("hybrid")[1]
|
182 |
+
s = s.replace("[", "").replace("]", "")
|
183 |
+
methods = s.split("+")
|
184 |
+
f0_computation_stack = []
|
185 |
+
|
186 |
+
print("Calculating f0 pitch estimations for methods: %s" % str(methods))
|
187 |
+
x = x.astype(np.float32)
|
188 |
+
x /= np.quantile(np.abs(x), 0.999)
|
189 |
+
# Get f0 calculations for all methods specified
|
190 |
+
for method in methods:
|
191 |
+
f0 = None
|
192 |
+
if method == "pm":
|
193 |
+
f0 = (
|
194 |
+
parselmouth.Sound(x, self.sr)
|
195 |
+
.to_pitch_ac(
|
196 |
+
time_step=time_step / 1000,
|
197 |
+
voicing_threshold=0.6,
|
198 |
+
pitch_floor=f0_min,
|
199 |
+
pitch_ceiling=f0_max,
|
200 |
+
)
|
201 |
+
.selected_array["frequency"]
|
202 |
+
)
|
203 |
+
pad_size = (p_len - len(f0) + 1) // 2
|
204 |
+
if pad_size > 0 or p_len - len(f0) - pad_size > 0:
|
205 |
+
f0 = np.pad(
|
206 |
+
f0, [[pad_size, p_len - len(f0) - pad_size]], mode="constant"
|
207 |
+
)
|
208 |
+
elif method == "crepe":
|
209 |
+
f0 = self.get_f0_official_crepe_computation(x, f0_min, f0_max)
|
210 |
+
f0 = f0[1:] # Get rid of extra first frame
|
211 |
+
elif method == "crepe-tiny":
|
212 |
+
f0 = self.get_f0_official_crepe_computation(x, f0_min, f0_max, "tiny")
|
213 |
+
f0 = f0[1:] # Get rid of extra first frame
|
214 |
+
elif method == "mangio-crepe":
|
215 |
+
f0 = self.get_f0_crepe_computation(
|
216 |
+
x, f0_min, f0_max, p_len, crepe_hop_length
|
217 |
+
)
|
218 |
+
elif method == "mangio-crepe-tiny":
|
219 |
+
f0 = self.get_f0_crepe_computation(
|
220 |
+
x, f0_min, f0_max, p_len, crepe_hop_length, "tiny"
|
221 |
+
)
|
222 |
+
elif method == "harvest":
|
223 |
+
f0 = cache_harvest_f0(input_audio_path, self.sr, f0_max, f0_min, 10)
|
224 |
+
if filter_radius > 2:
|
225 |
+
f0 = signal.medfilt(f0, 3)
|
226 |
+
f0 = f0[1:] # Get rid of first frame.
|
227 |
+
elif method == "dio": # Potentially buggy?
|
228 |
+
f0, t = pyworld.dio(
|
229 |
+
x.astype(np.double),
|
230 |
+
fs=self.sr,
|
231 |
+
f0_ceil=f0_max,
|
232 |
+
f0_floor=f0_min,
|
233 |
+
frame_period=10,
|
234 |
+
)
|
235 |
+
f0 = pyworld.stonemask(x.astype(np.double), f0, t, self.sr)
|
236 |
+
f0 = signal.medfilt(f0, 3)
|
237 |
+
f0 = f0[1:]
|
238 |
+
# elif method == "pyin": Not Working just yet
|
239 |
+
# f0 = self.get_f0_pyin_computation(x, f0_min, f0_max)
|
240 |
+
# Push method to the stack
|
241 |
+
f0_computation_stack.append(f0)
|
242 |
+
|
243 |
+
for fc in f0_computation_stack:
|
244 |
+
print(len(fc))
|
245 |
+
|
246 |
+
print("Calculating hybrid median f0 from the stack of: %s" % str(methods))
|
247 |
+
f0_median_hybrid = None
|
248 |
+
if len(f0_computation_stack) == 1:
|
249 |
+
f0_median_hybrid = f0_computation_stack[0]
|
250 |
+
else:
|
251 |
+
f0_median_hybrid = np.nanmedian(f0_computation_stack, axis=0)
|
252 |
+
return f0_median_hybrid
|
253 |
+
|
254 |
+
def get_f0(
|
255 |
+
self,
|
256 |
+
input_audio_path,
|
257 |
+
x,
|
258 |
+
p_len,
|
259 |
+
f0_up_key,
|
260 |
+
f0_method,
|
261 |
+
filter_radius,
|
262 |
+
crepe_hop_length,
|
263 |
+
inp_f0=None,
|
264 |
+
):
|
265 |
+
global input_audio_path2wav
|
266 |
+
time_step = self.window / self.sr * 1000
|
267 |
+
f0_min = 50
|
268 |
+
f0_max = 1100
|
269 |
+
f0_mel_min = 1127 * np.log(1 + f0_min / 700)
|
270 |
+
f0_mel_max = 1127 * np.log(1 + f0_max / 700)
|
271 |
+
if f0_method == "pm":
|
272 |
+
f0 = (
|
273 |
+
parselmouth.Sound(x, self.sr)
|
274 |
+
.to_pitch_ac(
|
275 |
+
time_step=time_step / 1000,
|
276 |
+
voicing_threshold=0.6,
|
277 |
+
pitch_floor=f0_min,
|
278 |
+
pitch_ceiling=f0_max,
|
279 |
+
)
|
280 |
+
.selected_array["frequency"]
|
281 |
+
)
|
282 |
+
pad_size = (p_len - len(f0) + 1) // 2
|
283 |
+
if pad_size > 0 or p_len - len(f0) - pad_size > 0:
|
284 |
+
f0 = np.pad(
|
285 |
+
f0, [[pad_size, p_len - len(f0) - pad_size]], mode="constant"
|
286 |
+
)
|
287 |
+
elif f0_method == "harvest":
|
288 |
+
input_audio_path2wav[input_audio_path] = x.astype(np.double)
|
289 |
+
f0 = cache_harvest_f0(input_audio_path, self.sr, f0_max, f0_min, 10)
|
290 |
+
if filter_radius > 2:
|
291 |
+
f0 = signal.medfilt(f0, 3)
|
292 |
+
elif f0_method == "dio": # Potentially Buggy?
|
293 |
+
f0, t = pyworld.dio(
|
294 |
+
x.astype(np.double),
|
295 |
+
fs=self.sr,
|
296 |
+
f0_ceil=f0_max,
|
297 |
+
f0_floor=f0_min,
|
298 |
+
frame_period=10,
|
299 |
+
)
|
300 |
+
f0 = pyworld.stonemask(x.astype(np.double), f0, t, self.sr)
|
301 |
+
f0 = signal.medfilt(f0, 3)
|
302 |
+
elif f0_method == "crepe":
|
303 |
+
f0 = self.get_f0_official_crepe_computation(x, f0_min, f0_max)
|
304 |
+
elif f0_method == "crepe-tiny":
|
305 |
+
f0 = self.get_f0_official_crepe_computation(x, f0_min, f0_max, "tiny")
|
306 |
+
elif f0_method == "mangio-crepe":
|
307 |
+
f0 = self.get_f0_crepe_computation(
|
308 |
+
x, f0_min, f0_max, p_len, crepe_hop_length
|
309 |
+
)
|
310 |
+
elif f0_method == "mangio-crepe-tiny":
|
311 |
+
f0 = self.get_f0_crepe_computation(
|
312 |
+
x, f0_min, f0_max, p_len, crepe_hop_length, "tiny"
|
313 |
+
)
|
314 |
+
elif f0_method == "rmvpe":
|
315 |
+
if hasattr(self, "model_rmvpe") == False:
|
316 |
+
from rmvpe import RMVPE
|
317 |
+
|
318 |
+
print("loading rmvpe model")
|
319 |
+
self.model_rmvpe = RMVPE(
|
320 |
+
"rmvpe.pt", is_half=self.is_half, device=self.device
|
321 |
+
)
|
322 |
+
f0 = self.model_rmvpe.infer_from_audio(x, thred=0.03)
|
323 |
+
|
324 |
+
elif "hybrid" in f0_method:
|
325 |
+
# Perform hybrid median pitch estimation
|
326 |
+
input_audio_path2wav[input_audio_path] = x.astype(np.double)
|
327 |
+
f0 = self.get_f0_hybrid_computation(
|
328 |
+
f0_method,
|
329 |
+
input_audio_path,
|
330 |
+
x,
|
331 |
+
f0_min,
|
332 |
+
f0_max,
|
333 |
+
p_len,
|
334 |
+
filter_radius,
|
335 |
+
crepe_hop_length,
|
336 |
+
time_step,
|
337 |
+
)
|
338 |
+
|
339 |
+
f0 *= pow(2, f0_up_key / 12)
|
340 |
+
# with open("test.txt","w")as f:f.write("\n".join([str(i)for i in f0.tolist()]))
|
341 |
+
tf0 = self.sr // self.window # 每秒f0点数
|
342 |
+
if inp_f0 is not None:
|
343 |
+
delta_t = np.round(
|
344 |
+
(inp_f0[:, 0].max() - inp_f0[:, 0].min()) * tf0 + 1
|
345 |
+
).astype("int16")
|
346 |
+
replace_f0 = np.interp(
|
347 |
+
list(range(delta_t)), inp_f0[:, 0] * 100, inp_f0[:, 1]
|
348 |
+
)
|
349 |
+
shape = f0[self.x_pad * tf0 : self.x_pad * tf0 + len(replace_f0)].shape[0]
|
350 |
+
f0[self.x_pad * tf0 : self.x_pad * tf0 + len(replace_f0)] = replace_f0[
|
351 |
+
:shape
|
352 |
+
]
|
353 |
+
# with open("test_opt.txt","w")as f:f.write("\n".join([str(i)for i in f0.tolist()]))
|
354 |
+
f0bak = f0.copy()
|
355 |
+
f0_mel = 1127 * np.log(1 + f0 / 700)
|
356 |
+
f0_mel[f0_mel > 0] = (f0_mel[f0_mel > 0] - f0_mel_min) * 254 / (
|
357 |
+
f0_mel_max - f0_mel_min
|
358 |
+
) + 1
|
359 |
+
f0_mel[f0_mel <= 1] = 1
|
360 |
+
f0_mel[f0_mel > 255] = 255
|
361 |
+
f0_coarse = np.rint(f0_mel).astype(np.int)
|
362 |
+
|
363 |
+
return f0_coarse, f0bak # 1-0
|
364 |
+
|
365 |
+
def vc(
|
366 |
+
self,
|
367 |
+
model,
|
368 |
+
net_g,
|
369 |
+
sid,
|
370 |
+
audio0,
|
371 |
+
pitch,
|
372 |
+
pitchf,
|
373 |
+
times,
|
374 |
+
index,
|
375 |
+
big_npy,
|
376 |
+
index_rate,
|
377 |
+
version,
|
378 |
+
protect,
|
379 |
+
): # ,file_index,file_big_npy
|
380 |
+
feats = torch.from_numpy(audio0)
|
381 |
+
if self.is_half:
|
382 |
+
feats = feats.half()
|
383 |
+
else:
|
384 |
+
feats = feats.float()
|
385 |
+
if feats.dim() == 2: # double channels
|
386 |
+
feats = feats.mean(-1)
|
387 |
+
assert feats.dim() == 1, feats.dim()
|
388 |
+
feats = feats.view(1, -1)
|
389 |
+
padding_mask = torch.BoolTensor(feats.shape).to(self.device).fill_(False)
|
390 |
+
|
391 |
+
inputs = {
|
392 |
+
"source": feats.to(self.device),
|
393 |
+
"padding_mask": padding_mask,
|
394 |
+
"output_layer": 9 if version == "v1" else 12,
|
395 |
+
}
|
396 |
+
t0 = ttime()
|
397 |
+
with torch.no_grad():
|
398 |
+
logits = model.extract_features(**inputs)
|
399 |
+
feats = model.final_proj(logits[0]) if version == "v1" else logits[0]
|
400 |
+
if protect < 0.5 and pitch != None and pitchf != None:
|
401 |
+
feats0 = feats.clone()
|
402 |
+
if (
|
403 |
+
isinstance(index, type(None)) == False
|
404 |
+
and isinstance(big_npy, type(None)) == False
|
405 |
+
and index_rate != 0
|
406 |
+
):
|
407 |
+
npy = feats[0].cpu().numpy()
|
408 |
+
if self.is_half:
|
409 |
+
npy = npy.astype("float32")
|
410 |
+
|
411 |
+
# _, I = index.search(npy, 1)
|
412 |
+
# npy = big_npy[I.squeeze()]
|
413 |
+
|
414 |
+
score, ix = index.search(npy, k=8)
|
415 |
+
weight = np.square(1 / score)
|
416 |
+
weight /= weight.sum(axis=1, keepdims=True)
|
417 |
+
npy = np.sum(big_npy[ix] * np.expand_dims(weight, axis=2), axis=1)
|
418 |
+
|
419 |
+
if self.is_half:
|
420 |
+
npy = npy.astype("float16")
|
421 |
+
feats = (
|
422 |
+
torch.from_numpy(npy).unsqueeze(0).to(self.device) * index_rate
|
423 |
+
+ (1 - index_rate) * feats
|
424 |
+
)
|
425 |
+
|
426 |
+
feats = F.interpolate(feats.permute(0, 2, 1), scale_factor=2).permute(0, 2, 1)
|
427 |
+
if protect < 0.5 and pitch != None and pitchf != None:
|
428 |
+
feats0 = F.interpolate(feats0.permute(0, 2, 1), scale_factor=2).permute(
|
429 |
+
0, 2, 1
|
430 |
+
)
|
431 |
+
t1 = ttime()
|
432 |
+
p_len = audio0.shape[0] // self.window
|
433 |
+
if feats.shape[1] < p_len:
|
434 |
+
p_len = feats.shape[1]
|
435 |
+
if pitch != None and pitchf != None:
|
436 |
+
pitch = pitch[:, :p_len]
|
437 |
+
pitchf = pitchf[:, :p_len]
|
438 |
+
|
439 |
+
if protect < 0.5 and pitch != None and pitchf != None:
|
440 |
+
pitchff = pitchf.clone()
|
441 |
+
pitchff[pitchf > 0] = 1
|
442 |
+
pitchff[pitchf < 1] = protect
|
443 |
+
pitchff = pitchff.unsqueeze(-1)
|
444 |
+
feats = feats * pitchff + feats0 * (1 - pitchff)
|
445 |
+
feats = feats.to(feats0.dtype)
|
446 |
+
p_len = torch.tensor([p_len], device=self.device).long()
|
447 |
+
with torch.no_grad():
|
448 |
+
if pitch != None and pitchf != None:
|
449 |
+
audio1 = (
|
450 |
+
(net_g.infer(feats, p_len, pitch, pitchf, sid)[0][0, 0])
|
451 |
+
.data.cpu()
|
452 |
+
.float()
|
453 |
+
.numpy()
|
454 |
+
)
|
455 |
+
else:
|
456 |
+
audio1 = (
|
457 |
+
(net_g.infer(feats, p_len, sid)[0][0, 0]).data.cpu().float().numpy()
|
458 |
+
)
|
459 |
+
del feats, p_len, padding_mask
|
460 |
+
if torch.cuda.is_available():
|
461 |
+
torch.cuda.empty_cache()
|
462 |
+
t2 = ttime()
|
463 |
+
times[0] += t1 - t0
|
464 |
+
times[2] += t2 - t1
|
465 |
+
return audio1
|
466 |
+
|
467 |
+
def pipeline(
|
468 |
+
self,
|
469 |
+
model,
|
470 |
+
net_g,
|
471 |
+
sid,
|
472 |
+
audio,
|
473 |
+
input_audio_path,
|
474 |
+
times,
|
475 |
+
f0_up_key,
|
476 |
+
f0_method,
|
477 |
+
file_index,
|
478 |
+
# file_big_npy,
|
479 |
+
index_rate,
|
480 |
+
if_f0,
|
481 |
+
filter_radius,
|
482 |
+
tgt_sr,
|
483 |
+
resample_sr,
|
484 |
+
rms_mix_rate,
|
485 |
+
version,
|
486 |
+
protect,
|
487 |
+
crepe_hop_length,
|
488 |
+
f0_file=None,
|
489 |
+
):
|
490 |
+
if (
|
491 |
+
file_index != ""
|
492 |
+
# and file_big_npy != ""
|
493 |
+
# and os.path.exists(file_big_npy) == True
|
494 |
+
and os.path.exists(file_index) == True
|
495 |
+
and index_rate != 0
|
496 |
+
):
|
497 |
+
try:
|
498 |
+
index = faiss.read_index(file_index)
|
499 |
+
# big_npy = np.load(file_big_npy)
|
500 |
+
big_npy = index.reconstruct_n(0, index.ntotal)
|
501 |
+
except:
|
502 |
+
traceback.print_exc()
|
503 |
+
index = big_npy = None
|
504 |
+
else:
|
505 |
+
index = big_npy = None
|
506 |
+
audio = signal.filtfilt(bh, ah, audio)
|
507 |
+
audio_pad = np.pad(audio, (self.window // 2, self.window // 2), mode="reflect")
|
508 |
+
opt_ts = []
|
509 |
+
if audio_pad.shape[0] > self.t_max:
|
510 |
+
audio_sum = np.zeros_like(audio)
|
511 |
+
for i in range(self.window):
|
512 |
+
audio_sum += audio_pad[i : i - self.window]
|
513 |
+
for t in range(self.t_center, audio.shape[0], self.t_center):
|
514 |
+
opt_ts.append(
|
515 |
+
t
|
516 |
+
- self.t_query
|
517 |
+
+ np.where(
|
518 |
+
np.abs(audio_sum[t - self.t_query : t + self.t_query])
|
519 |
+
== np.abs(audio_sum[t - self.t_query : t + self.t_query]).min()
|
520 |
+
)[0][0]
|
521 |
+
)
|
522 |
+
s = 0
|
523 |
+
audio_opt = []
|
524 |
+
t = None
|
525 |
+
t1 = ttime()
|
526 |
+
audio_pad = np.pad(audio, (self.t_pad, self.t_pad), mode="reflect")
|
527 |
+
p_len = audio_pad.shape[0] // self.window
|
528 |
+
inp_f0 = None
|
529 |
+
if hasattr(f0_file, "name") == True:
|
530 |
+
try:
|
531 |
+
with open(f0_file.name, "r") as f:
|
532 |
+
lines = f.read().strip("\n").split("\n")
|
533 |
+
inp_f0 = []
|
534 |
+
for line in lines:
|
535 |
+
inp_f0.append([float(i) for i in line.split(",")])
|
536 |
+
inp_f0 = np.array(inp_f0, dtype="float32")
|
537 |
+
except:
|
538 |
+
traceback.print_exc()
|
539 |
+
sid = torch.tensor(sid, device=self.device).unsqueeze(0).long()
|
540 |
+
pitch, pitchf = None, None
|
541 |
+
if if_f0 == 1:
|
542 |
+
pitch, pitchf = self.get_f0(
|
543 |
+
input_audio_path,
|
544 |
+
audio_pad,
|
545 |
+
p_len,
|
546 |
+
f0_up_key,
|
547 |
+
f0_method,
|
548 |
+
filter_radius,
|
549 |
+
crepe_hop_length,
|
550 |
+
inp_f0,
|
551 |
+
)
|
552 |
+
pitch = pitch[:p_len]
|
553 |
+
pitchf = pitchf[:p_len]
|
554 |
+
if self.device == "mps":
|
555 |
+
pitchf = pitchf.astype(np.float32)
|
556 |
+
pitch = torch.tensor(pitch, device=self.device).unsqueeze(0).long()
|
557 |
+
pitchf = torch.tensor(pitchf, device=self.device).unsqueeze(0).float()
|
558 |
+
t2 = ttime()
|
559 |
+
times[1] += t2 - t1
|
560 |
+
for t in opt_ts:
|
561 |
+
t = t // self.window * self.window
|
562 |
+
if if_f0 == 1:
|
563 |
+
audio_opt.append(
|
564 |
+
self.vc(
|
565 |
+
model,
|
566 |
+
net_g,
|
567 |
+
sid,
|
568 |
+
audio_pad[s : t + self.t_pad2 + self.window],
|
569 |
+
pitch[:, s // self.window : (t + self.t_pad2) // self.window],
|
570 |
+
pitchf[:, s // self.window : (t + self.t_pad2) // self.window],
|
571 |
+
times,
|
572 |
+
index,
|
573 |
+
big_npy,
|
574 |
+
index_rate,
|
575 |
+
version,
|
576 |
+
protect,
|
577 |
+
)[self.t_pad_tgt : -self.t_pad_tgt]
|
578 |
+
)
|
579 |
+
else:
|
580 |
+
audio_opt.append(
|
581 |
+
self.vc(
|
582 |
+
model,
|
583 |
+
net_g,
|
584 |
+
sid,
|
585 |
+
audio_pad[s : t + self.t_pad2 + self.window],
|
586 |
+
None,
|
587 |
+
None,
|
588 |
+
times,
|
589 |
+
index,
|
590 |
+
big_npy,
|
591 |
+
index_rate,
|
592 |
+
version,
|
593 |
+
protect,
|
594 |
+
)[self.t_pad_tgt : -self.t_pad_tgt]
|
595 |
+
)
|
596 |
+
s = t
|
597 |
+
if if_f0 == 1:
|
598 |
+
audio_opt.append(
|
599 |
+
self.vc(
|
600 |
+
model,
|
601 |
+
net_g,
|
602 |
+
sid,
|
603 |
+
audio_pad[t:],
|
604 |
+
pitch[:, t // self.window :] if t is not None else pitch,
|
605 |
+
pitchf[:, t // self.window :] if t is not None else pitchf,
|
606 |
+
times,
|
607 |
+
index,
|
608 |
+
big_npy,
|
609 |
+
index_rate,
|
610 |
+
version,
|
611 |
+
protect,
|
612 |
+
)[self.t_pad_tgt : -self.t_pad_tgt]
|
613 |
+
)
|
614 |
+
else:
|
615 |
+
audio_opt.append(
|
616 |
+
self.vc(
|
617 |
+
model,
|
618 |
+
net_g,
|
619 |
+
sid,
|
620 |
+
audio_pad[t:],
|
621 |
+
None,
|
622 |
+
None,
|
623 |
+
times,
|
624 |
+
index,
|
625 |
+
big_npy,
|
626 |
+
index_rate,
|
627 |
+
version,
|
628 |
+
protect,
|
629 |
+
)[self.t_pad_tgt : -self.t_pad_tgt]
|
630 |
+
)
|
631 |
+
audio_opt = np.concatenate(audio_opt)
|
632 |
+
if rms_mix_rate != 1:
|
633 |
+
audio_opt = change_rms(audio, 16000, audio_opt, tgt_sr, rms_mix_rate)
|
634 |
+
if resample_sr >= 16000 and tgt_sr != resample_sr:
|
635 |
+
audio_opt = librosa.resample(
|
636 |
+
audio_opt, orig_sr=tgt_sr, target_sr=resample_sr
|
637 |
+
)
|
638 |
+
audio_max = np.abs(audio_opt).max() / 0.99
|
639 |
+
max_int16 = 32768
|
640 |
+
if audio_max > 1:
|
641 |
+
max_int16 /= audio_max
|
642 |
+
audio_opt = (audio_opt * max_int16).astype(np.int16)
|
643 |
+
del pitch, pitchf, sid
|
644 |
+
if torch.cuda.is_available():
|
645 |
+
torch.cuda.empty_cache()
|
646 |
+
return audio_opt
|