hrtfpykit 0.1.1

A Python library for complete HRTF workflows, from SOFA file inspection and acoustic processing to visualization, comparison, and reproducible dataset pipelines construction.

What is hrtfpykit?

hrtfpykit is a Python library for complete Head Related Transfer Function (HRTF) workflows, from SOFA file inspection and acoustic processing to visualization, comparison, and reproducible dataset pipelines construction. It brings time/frequency domain handling, transformations, metrics, plots, and dataset pipelines into one workflow for research, HRTF individualization, and deep learning experiments.

Why hrtfpykit?

HRTF research often requires more than reading one SOFA file. If you have worked with HRTFs, you have probably met the usual ritual: searching for public datasets, discovering that every measurement setup has its own personality, adapting HRIR arrays to different dataset layouts, and moving between scripts, platforms, and tools with different assumptions. Datasets such as ARI, HUTUBS, and SONICOM made this work much more accessible, especially compared with the pre SOFA days of CSV files, spreadsheets, and heroic column name interpretation. Even today, the workflow can still become fragmented very quickly.

hrtfpykit was created to make those steps part of a clearer workflow. It gives researchers a way to work with HRTFs without losing the connection between the file, the acoustic representation, and the experiment.

What does hrtfpykit enable?

hrtfpykit can enable complete HRTF workflows, from file inspection to dataset construction. It is designed for users who need to understand, process, visualize, compare, and reuse HRTF data across research and deep learning tasks.

• Open, inspect, validate, edit, clone, and save SOFA files.
• Load HRTFs as acoustic objects with time domain and frequency domain views.
• Select source positions, ears, samples, and frequency bins.
• Modify HRTFs through transformations, domain conversions, and acoustic processing steps.
• Generate plots to inspect spectral cues, magnitude, amplitude, ITD, LSD, and differences between HRTFs with comparison plots, which is especially useful for HRTF individualization.
• Combine HRTFs with subject data such as anthropometry, metadata, meshes, and images.
• Create map-style dataset pipelines for training multimodal deep learning models.
• Build deep learning experiments for HRTF individualization and related tasks.

Architecture

hrtfpykit is organized around four public API entry points. The SOFA layer handles file structure, the HRTF layer builds the acoustic object, the plots layer visualizes HRTF data, and the datasets layer turns public HRTF resources into training and analysis samples.

• hrtfpykit.sofa: open, inspect, validate, edit, clone, and save SOFA files as structured Python objects.
• hrtfpykit.hrtf: load SOFA files as HRTF objects with IR data, TF data, source positions, transforms, metrics, and spherical harmonics.
• hrtfpykit.plots: visualize HRTF objects through spectral cues, source grids, binaural cues, spherical harmonic reconstructions, and comparison plots.
• hrtfpykit.datasets: build map-style public HRTF dataset pipelines with explicit inputs, targets, variants, splits, subject resources, custom resources, and batching utilities.

Guides

• Quick Start: first working examples for SOFA files, HRTF objects, plots, and datasets.
• Tutorials: guided notebook workflows for learning hrtfpykit step by step.

Installation

pip install hrtfpykit

For local installation from source:

git clone https://github.com/ArielAlvarez-Martinez/hrtfpykit.git
cd hrtfpykit
pip install .

For local development from source:

git clone https://github.com/ArielAlvarez-Martinez/hrtfpykit.git
cd hrtfpykit
pip install -e ".[test,docs]"

hrtfpykit requires Python 3.13 or newer.

License

hrtfpykit is distributed under the GPL 3.0 only license. See LICENSE for details.