| # Sortformer 一键说话人日志 (C++, 预编译可执行文件) | |
| # 用法: ./run_cpp_ax650.sh [input.wav] [output.rttm] [--fast] | |
| set -euo pipefail | |
| cd "$(dirname "$0")" | |
| export LD_LIBRARY_PATH=/soc/lib:${LD_LIBRARY_PATH:-} | |
| WAV="${1:-samples/sample_meeting.wav}" | |
| RTTM="${2:-$(basename "${WAV%.*}").rttm}" | |
| FAST=0 | |
| for arg in "$@"; do | |
| [[ "$arg" == "--fast" ]] && FAST=1 | |
| done | |
| if [[ "$FAST" == "1" ]]; then | |
| ENCODER=models/encoder_fifo40.axmodel | |
| EXTRA="--fifo-len 40 --spkcache-update-period 31" | |
| else | |
| ENCODER=models/encoder.axmodel | |
| EXTRA="" | |
| fi | |
| ./bin/sortformer_ax650 \ | |
| --preencode models/preencode.axmodel --encoder "$ENCODER" \ | |
| --wav "$WAV" --rttm "$RTTM" --threads 8 $EXTRA | |