scripts/infer_batch.py

"""Classify every .wav file in a directory.

Usage:
    pip install -r requirements.txt
    python infer_batch.py path/to/wav_dir
"""
import sys
from pathlib import Path

from gender_id import GenderClassifier

if __name__ == "__main__":
    if len(sys.argv) != 2:
        raise SystemExit(f"usage: python {sys.argv[0]} <wav_dir>")
    wav_dir = Path(sys.argv[1])
    clf = GenderClassifier()
    for path in sorted(wav_dir.glob("*.wav")):
        gender, confidence = clf.predict_file(str(path))
        print(f"{path.name}\t{gender}\t{confidence:.3f}")