wehub-resource-sync
20 lines
788 B
Python
20 lines
788 B
Python
import argparse
|
|
import functools
|
|
|
|
from mvector.trainer import MVectorTrainer
|
|
from mvector.utils.utils import add_arguments, print_arguments
|
|
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
add_arg = functools.partial(add_arguments, argparser=parser)
|
|
add_arg('configs', str, 'configs/cam++.yml', '配置文件')
|
|
add_arg('save_dir', str, 'dataset/features', '保存特征的路径')
|
|
add_arg('max_duration', int, 100, '提取特征的最大时长，避免过长显存不足，单位秒')
|
|
args = parser.parse_args()
|
|
print_arguments(args=args)
|
|
|
|
# 获取训练器
|
|
trainer = MVectorTrainer(configs=args.configs)
|
|
|
|
# 提取特征保存文件
|
|
trainer.extract_features(save_dir=args.save_dir, max_duration=args.max_duration)
|