jshiffer/Retrieval-based-Voice-Conversion
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

move cli

Browse Source
This commit is contained in:
Ftps
2024-01-21 11:59:32 +09:00
parent b29976b8f9
commit fb2960e336
7 changed files with 0 additions and 235 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
rvc/cli/cli.py
View File

@@ -1,33 +0,0 @@
import re
from typing import Optional, Pattern
import click
from rvc.cli.handler.infer import infer
from rvc.cli.handler.train import train
from rvc.cli.handler.uvr5 import uvr
from rvc.cli.utils.dlmodel import dlmodel
from rvc.cli.utils.env import env
from rvc.cli.utils.initialize import init
@click.group(
context_settings={"help_option_names": ["-h", "--help"]},
help="rvc cli feature list",
)
def cli():
pass
def main():
cli.add_command(infer)
cli.add_command(train)
cli.add_command(uvr)
cli.add_command(dlmodel)
cli.add_command(env)
cli.add_command(init)
cli()
if __name__ == "__main__":
main()

130
rvc/cli/handler/infer.py
View File

@@ -1,130 +0,0 @@
import logging
from pathlib import Path
import click
from dotenv import load_dotenv
from scipy.io import wavfile
from rvc.modules.vc.modules import VC
logging.getLogger("numba").setLevel(logging.WARNING)
@click.command(
context_settings={"help_option_names": ["-h", "--help"]},
help="inference audio",
)
@click.option(
"-m",
"--modelPath",
is_flag=False,
type=str,
help="Model path or filename (reads in the directory set in env)",
required=True,
)
@click.option(
"-i",
"--inputPath",
is_flag=False,
type=Path,
help="input audio path or folder",
required=True,
)
@click.option(
"-o",
"--outputPath",
is_flag=False,
type=Path,
help="output audio path or folder",
required=True,
)
@click.option(
"-s", "--sid", is_flag=False, type=int, help="Speaker/Singer id", default=0
)
@click.option("-fu", "--f0upkey", is_flag=False, type=int, help="Transpose", default=0)
@click.option(
"-fm",
"--f0method",
is_flag=False,
type=str,
help="Pitch extraction algorith",
default="rmvpe",
)
@click.option(
"-ff", "--f0file", is_flag=False, type=Path, help="F0 curve file (optional)"
)
@click.option("-if", "--indexFile", is_flag=False, type=Path, help="Feature index file")
@click.option(
"-ir",
"--indexRate",
is_flag=False,
type=float,
help="Search feature ratio",
default=0.75,
)
@click.option(
"-fr",
"--filterRadius",
is_flag=False,
type=int,
help="Apply median filtering",
default=3,
)
@click.option(
"-rsr",
"--resamplesr",
is_flag=False,
type=int,
help="Resample the output audio",
default=0,
)
@click.option(
"-rmr",
"--rmsmixrate",
is_flag=False,
type=float,
help="Adjust the volume envelope scaling",
default=0.25,
)
@click.option(
"-p",
"--protect",
is_flag=False,
type=float,
help="Protect voiceless consonants and breath sounds",
default=0.33,
)
def infer(
modelpath,
inputpath,
outputpath,
sid,
f0upkey,
f0method,
f0file,
indexfile,
indexrate,
filterradius,
resamplesr,
rmsmixrate,
protect,
):
load_dotenv()
vc = VC()
vc.get_vc(modelpath)
tgt_sr, audio_opt, times, _ = vc.vc_single(
sid,
inputpath,
f0upkey,
f0method,
f0file,
indexfile,
indexrate,
filterradius,
resamplesr,
rmsmixrate,
protect,
)
wavfile.write(outputpath, tgt_sr, audio_opt)
click.echo(times)
click.echo(f"Finish inference. Check {outputpath}")

6
rvc/cli/handler/train.py
View File

@@ -1,6 +0,0 @@
import click
@click.command()
def train():
pass

6
rvc/cli/handler/uvr5.py
View File

@@ -1,6 +0,0 @@
import click
@click.command()
def uvr():
pass

9
rvc/cli/utils/dlmodel.py
View File

@@ -1,9 +0,0 @@
import urllib
import click
@click.command()
def dlmodel() -> None:
# Download models [harvest, uvr5, and more ]
pass

40
rvc/cli/utils/env.py
View File

@@ -1,40 +0,0 @@
"""
setup or cleanup enviroment file
usage: rvc env [create / cleanup]
Default: [nowDir/.env]
"""
import os
import click
@click.group()
def env():
pass
@env.command()
def create():
env_file_path = os.path.join(os.getcwd(), ".env")
if not os.path.exists(env_file_path):
default_values = {
"weight_root": "",
"weight_uvr5_root": "",
"index_root": "",
"rmvpe_root": "",
"hubert_path": "",
"save_uvr_path": "",
"TEMP": "",
"pretrained": "",
}
with open(env_file_path, "w") as env_file:
for key, value in default_values.items():
env_file.write(f"{key}={value}\n")
click.echo(f"{env_file_path} created successfully.")
else:
click.echo(f"{env_file_path} already exists, no change")

11
rvc/cli/utils/initialize.py
View File

@@ -1,11 +0,0 @@
"""
Uage: rvc init
download model and setup environmmnt file
"""
import click
@click.command()
def init():
pass