File size: 1,143 Bytes
5e23710 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | #!/usr/bin/env python3
# coding: utf-8
"""GCRN AX650 语音增强示例：noisy.wav -> enhanced.wav"""
import argparse
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parent))
from gcrn_sdk import GCRNDenoiser
def main() -> None:
ap = argparse.ArgumentParser(description="GCRN speech enhancement on AX650")
ap.add_argument("--model", default="models/model.axmodel", help="axmodel (或 onnx)")
ap.add_argument("--input", default="test_audio/mix.wav", help="16kHz mono noisy wav")
ap.add_argument("--output", default="enhanced.wav", help="输出 wav")
ap.add_argument("--report", default="gcrn_report.json", help="性能报告输出")
args = ap.parse_args()
denoiser = GCRNDenoiser(args.model)
report = denoiser.enhance_file(args.input, args.output)
denoiser.save_report(report, args.report)
print(f"input={args.input}")
print(f"output={args.output}")
print(
f"elapsed={report['elapsed_seconds']:.3f}s "
f"audio={report['audio_seconds']:.3f}s "
f"rtf={report['real_time_factor']:.4f}"
)
if __name__ == "__main__":
main()
|