init
This commit is contained in:
41
scripts/run_a2f.py
Normal file
41
scripts/run_a2f.py
Normal file
@ -0,0 +1,41 @@
|
||||
import subprocess
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
# ========= 基础路径 =========
|
||||
BASE_DIR = Path(__file__).parent.parent
|
||||
|
||||
A2F_SCRIPT = BASE_DIR / "external" / "Audio2Face-3D-Samples" / "scripts" / "audio2face_3d_microservices_interaction_app" / "a2f_3d.py"
|
||||
AUDIO_FILE = BASE_DIR / "data" / "input" / "audio" / "Mark_joy.wav"
|
||||
CONFIG_FILE = BASE_DIR / "external" / "Audio2Face-3D-Samples" / "scripts" / "audio2face_3d_microservices_interaction_app" / "config" / "config_james.yml"
|
||||
|
||||
A2F_URL = "192.168.1.39:52000"
|
||||
|
||||
# ========= 调用命令 =========
|
||||
cmd = [
|
||||
sys.executable, # 使用当前 venv 的 python
|
||||
str(A2F_SCRIPT),
|
||||
"run_inference",
|
||||
str(AUDIO_FILE),
|
||||
str(CONFIG_FILE),
|
||||
"--url",
|
||||
A2F_URL
|
||||
]
|
||||
|
||||
print("Running Audio2Face-3D inference...")
|
||||
print("Command:", " ".join(cmd))
|
||||
|
||||
result = subprocess.run(
|
||||
cmd,
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.STDOUT,
|
||||
text=True
|
||||
)
|
||||
|
||||
print("=== A2F OUTPUT ===")
|
||||
print(result.stdout)
|
||||
|
||||
if result.returncode != 0:
|
||||
raise RuntimeError("Audio2Face-3D inference failed")
|
||||
|
||||
print("Inference finished successfully.")
|
||||
Reference in New Issue
Block a user