73 lines
2.4 KiB
Python
73 lines
2.4 KiB
Python
import argparse
|
|
import os
|
|
import sys
|
|
|
|
print("Command-line arguments:", sys.argv)
|
|
|
|
now_dir = os.getcwd()
|
|
sys.path.append(now_dir)
|
|
import sys
|
|
|
|
import tqdm as tq
|
|
from dotenv import load_dotenv
|
|
from scipy.io import wavfile
|
|
|
|
from configs.config import Config
|
|
from infer.modules.vc.modules import VC
|
|
|
|
|
|
def arg_parse() -> tuple:
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--f0up_key", type=int, default=0)
|
|
parser.add_argument("--input_path", type=str, help="input path")
|
|
parser.add_argument("--index_path", type=str, help="index path")
|
|
parser.add_argument("--f0method", type=str, default="harvest", help="harvest or pm")
|
|
parser.add_argument("--opt_path", type=str, help="opt path")
|
|
parser.add_argument("--model_name", type=str, help="store in assets/weight_root")
|
|
parser.add_argument("--index_rate", type=float, default=0.66, help="index rate")
|
|
parser.add_argument("--device", type=str, help="device")
|
|
parser.add_argument("--is_half", type=bool, help="use half -> True")
|
|
parser.add_argument("--filter_radius", type=int, default=3, help="filter radius")
|
|
parser.add_argument("--resample_sr", type=int, default=0, help="resample sr")
|
|
parser.add_argument("--rms_mix_rate", type=float, default=1, help="rms mix rate")
|
|
parser.add_argument("--protect", type=float, default=0.33, help="protect")
|
|
|
|
args = parser.parse_args()
|
|
sys.argv = sys.argv[:1]
|
|
|
|
return args
|
|
|
|
|
|
def main():
|
|
load_dotenv()
|
|
args = arg_parse()
|
|
config = Config()
|
|
config.device = args.device if args.device else config.device
|
|
config.is_half = args.is_half if args.is_half else config.is_half
|
|
vc = VC(config)
|
|
vc.get_vc(args.model_name)
|
|
audios = os.listdir(args.input_path)
|
|
for file in tq.tqdm(audios):
|
|
if file.endswith(".wav"):
|
|
file_path = os.path.join(args.input_path, file)
|
|
_, wav_opt = vc.vc_single(
|
|
0,
|
|
file_path,
|
|
args.f0up_key,
|
|
None,
|
|
args.f0method,
|
|
args.index_path,
|
|
None,
|
|
args.index_rate,
|
|
args.filter_radius,
|
|
args.resample_sr,
|
|
args.rms_mix_rate,
|
|
args.protect,
|
|
)
|
|
out_path = os.path.join(args.opt_path, file)
|
|
wavfile.write(out_path, wav_opt[0], wav_opt[1])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|