Skip to content

Latest commit

 

History

History
72 lines (52 loc) · 2.57 KB

README.md

File metadata and controls

72 lines (52 loc) · 2.57 KB
Raw

Time-of-arrival estimation and phase unwrapping of HRTF with integer linear programming

Source code of the papers Time-of-arrival estimation and phase unwrapping of head-related transfer functions with integer linear programming (AES Convention 156) and Arbitrarily sampled signal reconstruction using relative difference features (GSP Workshop 2024).

Getting started

Please install the required packages by running the following command. You'll need to have Python >=3.11 installed on your system.

pip install -r requirements.txt

Preprocessing

Given a single HRTF sofa file, preprocess.py will compute its TOAs using 36 different configurations we stated in the paper and store the results into a folder.

python preprocess.py input.sofa output_folder --toa-weight 0.1 --oversampling 10

Here, --toa-weight controls the value of $\lambda$ and $w_{\delta, i}$ of exponential weighting in the paper. The output npz files has the format of {edgeslist/ilp/l2}_toa_{True/False}_cross_{True/False}_{angle/dot/none}.npz. They corresponds to the terms in the paper as follows:

Name Name in ther paper
edgeslist EDGY
ilp SIMP
l2 LS
toa_True w/ Min.
toa_False w/o Min.
cross_True w/ Cross
cross_False w/o Cross
angle EXP
dot CORR
none NONE

Experiments

ITD/aligned-HRIRs reconstruction

Please create a folder named processed and put the preprocessed directories in it. Then, execute the notebook eval to evaluate the estimated TOAs.

Noise robustness

Please execute the notebook noise-robust to see the results. The notebook includes the code to generate Figure 2 in the paper. Changing the sofa file to the one you want to test will generate the results for the given HRTF.

Phase unwrapping

Please execute the notebook phase-unwrapping to see the results. The notebook includes the code to generate Figure 4 and 5 in the paper. The notebook uses kamui for ILP-based unwrapping.

Citation

If you use our work in your research, please cite it with the following BibTex entry:

@misc{ycy2024hrtf,
    title={Time-of-arrival Estimation and Phase Unwrapping of Head-related Transfer Functions With Integer Linear Programming},
    author={Chin-Yun Yu and Johan Pauwels and György Fazekas},
    year={2024},
    eprint={2405.06804},
    archivePrefix={arXiv},
    primaryClass={cs.SD}
}