kwau/sovits-isla

1import argparse2import os3import pickle4 5import utils6 7if __name__ == "__main__":8    parser = argparse.ArgumentParser()9    parser.add_argument(10        "--root_dir", type=str, default="dataset/44k", help="path to root dir"11    )12    parser.add_argument('-c', '--config', type=str, default="./configs/config.json",13                    help='JSON file for configuration')14    parser.add_argument(15        "--output_dir", type=str, default="logs/44k", help="path to output dir"16    )17 18    args = parser.parse_args()19 20    hps = utils.get_hparams_from_file(args.config)21    spk_dic = hps.spk22    result = {}23    24    for k,v in spk_dic.items():25        print(f"now, index {k} feature...")26        index = utils.train_index(k,args.root_dir)27        result[v] = index28 29    with open(os.path.join(args.output_dir,"feature_and_index.pkl"),"wb") as f:30        pickle.dump(result,f)