File size: 1,149 Bytes
c0da797 | 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 | #!/usr/bin/env python3
"""OpusBWE.AXERA demo: 16 kHz wav -> 48 kHz wav.
The neural feature net runs on the AX650 NPU; the DSP synthesis is the bundled
C++ backend. This wrapper starts `bin/bwe_ax650` with the right environment.
"""
import argparse
import os
import subprocess
import sys
ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
def main():
ap = argparse.ArgumentParser(description="OpusBWE.AXERA 16k -> 48k bandwidth extension")
ap.add_argument("input", help="input wav (16 kHz)")
ap.add_argument("output", help="output wav (48 kHz)")
ap.add_argument("--model", default=os.path.join(ROOT, "models", "bwe_feature_net.axmodel"))
ap.add_argument("--bin", default=os.path.join(ROOT, "bin", "bwe_ax650"))
args = ap.parse_args()
if not os.path.isfile(args.bin):
sys.exit(f"backend not found: {args.bin}")
env = dict(os.environ)
env["LD_LIBRARY_PATH"] = "/soc/lib:" + env.get("LD_LIBRARY_PATH", "")
cmd = [args.bin, args.model, args.input, args.output]
print("+", " ".join(cmd))
sys.exit(subprocess.run(cmd, env=env).returncode)
if __name__ == "__main__":
main()
|