2023-05-31 18:45:01 +00:00
|
|
|
import argparse
|
2023-06-26 06:57:53 +00:00
|
|
|
import os
|
|
|
|
|
import pickle
|
2023-05-31 18:45:01 +00:00
|
|
|
|
2023-06-26 06:57:53 +00:00
|
|
|
import utils
|
2023-05-31 18:45:01 +00:00
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
parser = argparse.ArgumentParser()
|
|
|
|
|
parser.add_argument(
|
|
|
|
|
"--root_dir", type=str, default="dataset/44k", help="path to root dir"
|
|
|
|
|
)
|
|
|
|
|
parser.add_argument('-c', '--config', type=str, default="./configs/config.json",
|
|
|
|
|
help='JSON file for configuration')
|
|
|
|
|
parser.add_argument(
|
|
|
|
|
"--output_dir", type=str, default="logs/44k", help="path to output dir"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
|
|
|
|
hps = utils.get_hparams_from_file(args.config)
|
|
|
|
|
spk_dic = hps.spk
|
|
|
|
|
result = {}
|
|
|
|
|
|
|
|
|
|
for k,v in spk_dic.items():
|
|
|
|
|
print(f"now, index {k} feature...")
|
|
|
|
|
index = utils.train_index(k,args.root_dir)
|
|
|
|
|
result[v] = index
|
|
|
|
|
|
|
|
|
|
with open(os.path.join(args.output_dir,"feature_and_index.pkl"),"wb") as f:
|
|
|
|
|
pickle.dump(result,f)
|
