OPENGIT/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git synced 2025-03-10 15:54:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

rewrite _get_name_params

Browse Source
This commit is contained in:
Ftps 2023-04-15 02:16:53 +09:00
parent 40dafd9c42
commit 3297f1e518
2 changed files with 287 additions and 158 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

263
uvr5_pack/name_params.json Normal file
View File

@ -0,0 +1,263 @@
{
"equivalent" : [
{
"model_hash_name" : [
{
"hash_name": "47939caf0cfe52a0e81442b85b971dfd",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_44100.json",
"param_name": "4band_44100"
},
{
"hash_name": "4e4ecb9764c50a8c414fee6e10395bbe",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_v2.json",
"param_name": "4band_v2"
},
{
"hash_name": "ca106edd563e034bde0bdec4bb7a4b36",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_v2.json",
"param_name": "4band_v2"
},
{
"hash_name": "e60a1e84803ce4efc0a6551206cc4b71",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_44100.json",
"param_name": "4band_44100"
},
{
"hash_name": "a82f14e75892e55e994376edbf0c8435",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_44100.json",
"param_name": "4band_44100"
},
{
"hash_name": "6dd9eaa6f0420af9f1d403aaafa4cc06",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_v2_sn.json",
"param_name": "4band_v2_sn"
},
{
"hash_name": "08611fb99bd59eaa79ad27c58d137727",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_v2_sn.json",
"param_name": "4band_v2_sn"
},
{
"hash_name": "5c7bbca45a187e81abbbd351606164e5",
"model_params": "uvr5_pack/lib_v5/modelparams/3band_44100_msb2.json",
"param_name": "3band_44100_msb2"
},
{
"hash_name": "d6b2cb685a058a091e5e7098192d3233",
"model_params": "uvr5_pack/lib_v5/modelparams/3band_44100_msb2.json",
"param_name": "3band_44100_msb2"
},
{
"hash_name": "c1b9f38170a7c90e96f027992eb7c62b",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_44100.json",
"param_name": "4band_44100"
},
{
"hash_name": "c3448ec923fa0edf3d03a19e633faa53",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_44100.json",
"param_name": "4band_44100"
},
{
"hash_name": "68aa2c8093d0080704b200d140f59e54",
"model_params": "uvr5_pack/lib_v5/modelparams/3band_44100.json",
"param_name": "3band_44100"
},
{
"hash_name": "fdc83be5b798e4bd29fe00fe6600e147",
"model_params": "uvr5_pack/lib_v5/modelparams/3band_44100_mid.json",
"param_name": "3band_44100_mid.json"
},
{
"hash_name": "2ce34bc92fd57f55db16b7a4def3d745",
"model_params": "uvr5_pack/lib_v5/modelparams/3band_44100_mid.json",
"param_name": "3band_44100_mid.json"
},
{
"hash_name": "52fdca89576f06cf4340b74a4730ee5f",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_44100.json",
"param_name": "4band_44100.json"
},
{
"hash_name": "41191165b05d38fc77f072fa9e8e8a30",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_44100.json",
"param_name": "4band_44100.json"
},
{
"hash_name": "89e83b511ad474592689e562d5b1f80e",
"model_params": "uvr5_pack/lib_v5/modelparams/2band_32000.json",
"param_name": "2band_32000.json"
},
{
"hash_name": "0b954da81d453b716b114d6d7c95177f",
"model_params": "uvr5_pack/lib_v5/modelparams/2band_32000.json",
"param_name": "2band_32000.json"
}
],
"v4 Models": [
{
"hash_name": "6a00461c51c2920fd68937d4609ed6c8",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr16000_hl512.json",
"param_name": "1band_sr16000_hl512"
},
{
"hash_name": "0ab504864d20f1bd378fe9c81ef37140",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr32000_hl512.json",
"param_name": "1band_sr32000_hl512"
},
{
"hash_name": "7dd21065bf91c10f7fccb57d7d83b07f",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr32000_hl512.json",
"param_name": "1band_sr32000_hl512"
},
{
"hash_name": "80ab74d65e515caa3622728d2de07d23",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr32000_hl512.json",
"param_name": "1band_sr32000_hl512"
},
{
"hash_name": "edc115e7fc523245062200c00caa847f",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr33075_hl384.json",
"param_name": "1band_sr33075_hl384"
},
{
"hash_name": "28063e9f6ab5b341c5f6d3c67f2045b7",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr33075_hl384.json",
"param_name": "1band_sr33075_hl384"
},
{
"hash_name": "b58090534c52cbc3e9b5104bad666ef2",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr44100_hl512.json",
"param_name": "1band_sr44100_hl512"
},
{
"hash_name": "0cdab9947f1b0928705f518f3c78ea8f",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr44100_hl512.json",
"param_name": "1band_sr44100_hl512"
},
{
"hash_name": "ae702fed0238afb5346db8356fe25f13",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr44100_hl1024.json",
"param_name": "1band_sr44100_hl1024"
}
]
}
],
"User Models" : [
{
"1 Band": [
{
"hash_name": "1band_sr16000_hl512",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr16000_hl512.json",
"param_name": "1band_sr16000_hl512"
},
{
"hash_name": "1band_sr32000_hl512",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr32000_hl512.json",
"param_name": "1band_sr16000_hl512"
},
{
"hash_name": "1band_sr33075_hl384",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr33075_hl384.json",
"param_name": "1band_sr33075_hl384"
},
{
"hash_name": "1band_sr44100_hl256",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr44100_hl256.json",
"param_name": "1band_sr44100_hl256"
},
{
"hash_name": "1band_sr44100_hl512",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr44100_hl512.json",
"param_name": "1band_sr44100_hl512"
},
{
"hash_name": "1band_sr44100_hl1024",
"model_params": "uvr5_pack/lib_v5/modelparams/1band_sr44100_hl1024.json",
"param_name": "1band_sr44100_hl1024"
}
],
"2 Band": [
{
"hash_name": "2band_44100_lofi",
"model_params": "uvr5_pack/lib_v5/modelparams/2band_44100_lofi.json",
"param_name": "2band_44100_lofi"
},
{
"hash_name": "2band_32000",
"model_params": "uvr5_pack/lib_v5/modelparams/2band_32000.json",
"param_name": "2band_32000"
},
{
"hash_name": "2band_48000",
"model_params": "uvr5_pack/lib_v5/modelparams/2band_48000.json",
"param_name": "2band_48000"
}
],
"3 Band": [
{
"hash_name": "3band_44100",
"model_params": "uvr5_pack/lib_v5/modelparams/3band_44100.json",
"param_name": "3band_44100"
},
{
"hash_name": "3band_44100_mid",
"model_params": "uvr5_pack/lib_v5/modelparams/3band_44100_mid.json",
"param_name": "3band_44100_mid"
},
{
"hash_name": "3band_44100_msb2",
"model_params": "uvr5_pack/lib_v5/modelparams/3band_44100_msb2.json",
"param_name": "3band_44100_msb2"
}
],
"4 Band": [
{
"hash_name": "4band_44100",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_44100.json",
"param_name": "4band_44100"
},
{
"hash_name": "4band_44100_mid",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_44100_mid.json",
"param_name": "4band_44100_mid"
},
{
"hash_name": "4band_44100_msb",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_44100_msb.json",
"param_name": "4band_44100_msb"
},
{
"hash_name": "4band_44100_msb2",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_44100_msb2.json",
"param_name": "4band_44100_msb2"
},
{
"hash_name": "4band_44100_reverse",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_44100_reverse.json",
"param_name": "4band_44100_reverse"
},
{
"hash_name": "4band_44100_sw",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_44100_sw.json",
"param_name": "4band_44100_sw"
},
{
"hash_name": "4band_v2",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_v2.json",
"param_name": "4band_v2"
},
{
"hash_name": "4band_v2_sn",
"model_params": "uvr5_pack/lib_v5/modelparams/4band_v2_sn.json",
"param_name": "4band_v2_sn"
},
{
"hash_name": "tmodelparam",
"model_params": "uvr5_pack/lib_v5/modelparams/tmodelparam.json",
"param_name": "User Model Param Set"
}
]
}
]
}

182
uvr5_pack/utils.py
View File

@ -1,6 +1,14 @@
import torch import torch
import numpy as np import numpy as np
from tqdm import tqdm from tqdm import tqdm
import json
def load_data(file_name: str = "./uvr5_pack/data.json") -> dict:
with open(file_name, "r") as f:
data = json.load(f)
return data
def make_padding(width, cropsize, offset): def make_padding(width, cropsize, offset):
@ -91,164 +99,22 @@ def inference(X_spec, device, model, aggressiveness, data):
def _get_name_params(model_path, model_hash): def _get_name_params(model_path, model_hash):
data = load_data()
flag = False
ModelName = model_path ModelName = model_path
if model_hash == "47939caf0cfe52a0e81442b85b971dfd": for type in list(data):
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_44100.json") for model in list(data[type][0]):
param_name_auto = str("4band_44100") for i in range(len(data[type][0][model])):
if model_hash == "4e4ecb9764c50a8c414fee6e10395bbe": if str(data[type][0][model][i]["hash_name"]) == model_hash:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_v2.json") flag = True
param_name_auto = str("4band_v2") elif str(data[type][0][model][i]["hash_name"]) in ModelName:
if model_hash == "ca106edd563e034bde0bdec4bb7a4b36": flag = True
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_v2.json")
param_name_auto = str("4band_v2")
if model_hash == "e60a1e84803ce4efc0a6551206cc4b71":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_44100.json")
param_name_auto = str("4band_44100")
if model_hash == "a82f14e75892e55e994376edbf0c8435":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_44100.json")
param_name_auto = str("4band_44100")
if model_hash == "6dd9eaa6f0420af9f1d403aaafa4cc06":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_v2_sn.json")
param_name_auto = str("4band_v2_sn")
if model_hash == "08611fb99bd59eaa79ad27c58d137727":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_v2_sn.json")
param_name_auto = str("4band_v2_sn")
if model_hash == "5c7bbca45a187e81abbbd351606164e5":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/3band_44100_msb2.json")
param_name_auto = str("3band_44100_msb2")
if model_hash == "d6b2cb685a058a091e5e7098192d3233":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/3band_44100_msb2.json")
param_name_auto = str("3band_44100_msb2")
if model_hash == "c1b9f38170a7c90e96f027992eb7c62b":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_44100.json")
param_name_auto = str("4band_44100")
if model_hash == "c3448ec923fa0edf3d03a19e633faa53":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_44100.json")
param_name_auto = str("4band_44100")
if model_hash == "68aa2c8093d0080704b200d140f59e54":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/3band_44100.json")
param_name_auto = str("3band_44100.json")
if model_hash == "fdc83be5b798e4bd29fe00fe6600e147":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/3band_44100_mid.json")
param_name_auto = str("3band_44100_mid.json")
if model_hash == "2ce34bc92fd57f55db16b7a4def3d745":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/3band_44100_mid.json")
param_name_auto = str("3band_44100_mid.json")
if model_hash == "52fdca89576f06cf4340b74a4730ee5f":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_44100.json")
param_name_auto = str("4band_44100.json")
if model_hash == "41191165b05d38fc77f072fa9e8e8a30":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_44100.json")
param_name_auto = str("4band_44100.json")
if model_hash == "89e83b511ad474592689e562d5b1f80e":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/2band_32000.json")
param_name_auto = str("2band_32000.json")
if model_hash == "0b954da81d453b716b114d6d7c95177f":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/2band_32000.json")
param_name_auto = str("2band_32000.json")
# v4 Models if flag:
if model_hash == "6a00461c51c2920fd68937d4609ed6c8": model_params_auto = data[type][0][model][i]["model_params"]
model_params_auto = str("uvr5_pack/lib_v5/modelparams/1band_sr16000_hl512.json") param_name_auto = data[type][0][model][i]["param_name"]
param_name_auto = str("1band_sr16000_hl512") if type == "equivalent":
if model_hash == "0ab504864d20f1bd378fe9c81ef37140": return param_name_auto, model_params_auto
model_params_auto = str("uvr5_pack/lib_v5/modelparams/1band_sr32000_hl512.json") else:
param_name_auto = str("1band_sr32000_hl512") flag = False
if model_hash == "7dd21065bf91c10f7fccb57d7d83b07f":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/1band_sr32000_hl512.json")
param_name_auto = str("1band_sr32000_hl512")
if model_hash == "80ab74d65e515caa3622728d2de07d23":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/1band_sr32000_hl512.json")
param_name_auto = str("1band_sr32000_hl512")
if model_hash == "edc115e7fc523245062200c00caa847f":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/1band_sr33075_hl384.json")
param_name_auto = str("1band_sr33075_hl384")
if model_hash == "28063e9f6ab5b341c5f6d3c67f2045b7":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/1band_sr33075_hl384.json")
param_name_auto = str("1band_sr33075_hl384")
if model_hash == "b58090534c52cbc3e9b5104bad666ef2":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/1band_sr44100_hl512.json")
param_name_auto = str("1band_sr44100_hl512")
if model_hash == "0cdab9947f1b0928705f518f3c78ea8f":
model_params_auto = str("uvr5_pack/lib_v5/modelparams/1band_sr44100_hl512.json")
param_name_auto = str("1band_sr44100_hl512")
if model_hash == "ae702fed0238afb5346db8356fe25f13":
model_params_auto = str(
"uvr5_pack/lib_v5/modelparams/1band_sr44100_hl1024.json"
)
param_name_auto = str("1band_sr44100_hl1024")
# User Models
# 1 Band
if "1band_sr16000_hl512" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/1band_sr16000_hl512.json")
param_name_auto = str("1band_sr16000_hl512")
if "1band_sr32000_hl512" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/1band_sr32000_hl512.json")
param_name_auto = str("1band_sr32000_hl512")
if "1band_sr33075_hl384" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/1band_sr33075_hl384.json")
param_name_auto = str("1band_sr33075_hl384")
if "1band_sr44100_hl256" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/1band_sr44100_hl256.json")
param_name_auto = str("1band_sr44100_hl256")
if "1band_sr44100_hl512" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/1band_sr44100_hl512.json")
param_name_auto = str("1band_sr44100_hl512")
if "1band_sr44100_hl1024" in ModelName:
model_params_auto = str(
"uvr5_pack/lib_v5/modelparams/1band_sr44100_hl1024.json"
)
param_name_auto = str("1band_sr44100_hl1024")
# 2 Band
if "2band_44100_lofi" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/2band_44100_lofi.json")
param_name_auto = str("2band_44100_lofi")
if "2band_32000" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/2band_32000.json")
param_name_auto = str("2band_32000")
if "2band_48000" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/2band_48000.json")
param_name_auto = str("2band_48000")
# 3 Band
if "3band_44100" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/3band_44100.json")
param_name_auto = str("3band_44100")
if "3band_44100_mid" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/3band_44100_mid.json")
param_name_auto = str("3band_44100_mid")
if "3band_44100_msb2" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/3band_44100_msb2.json")
param_name_auto = str("3band_44100_msb2")
# 4 Band
if "4band_44100" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_44100.json")
param_name_auto = str("4band_44100")
if "4band_44100_mid" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_44100_mid.json")
param_name_auto = str("4band_44100_mid")
if "4band_44100_msb" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_44100_msb.json")
param_name_auto = str("4band_44100_msb")
if "4band_44100_msb2" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_44100_msb2.json")
param_name_auto = str("4band_44100_msb2")
if "4band_44100_reverse" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_44100_reverse.json")
param_name_auto = str("4band_44100_reverse")
if "4band_44100_sw" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_44100_sw.json")
param_name_auto = str("4band_44100_sw")
if "4band_v2" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_v2.json")
param_name_auto = str("4band_v2")
if "4band_v2_sn" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/4band_v2_sn.json")
param_name_auto = str("4band_v2_sn")
if "tmodelparam" in ModelName:
model_params_auto = str("uvr5_pack/lib_v5/modelparams/tmodelparam.json")
param_name_auto = str("User Model Param Set")
return param_name_auto, model_params_auto return param_name_auto, model_params_auto