FAST-RIR: FAST NEURAL DIFFUSE ROOM IMPULSE RESPONSE GENERATOR
This is the official implementation of our neural-network-based fast diffuse room impulse response generator (FAST-RIR) for generating roomimpulse responses (RIRs) for a given rectangular acoustic environment. Our model is inspired by StackGAN architecture. The audio examples and spectrograms of the generated RIRs are available here.
Requirements
Python3.6
Pytorch
python-dateutil
easydict
pandas
torchfile
gdown
pickle
Embedding
Each normalized embedding is created as follows: If you are using our trained model, you may need to use extra parameter Correction(CRR).
Listener Position = LP
Source Position = SP
Room Dimension = RD
Reverberation Time = T60
Correction = CRR
CRR = 0.1 if 0.5
<0.6
CRR = 0.2 if T60>0.6
CRR = 0 otherwise
Embedding = ([LP_X,LP_Y,LP_Z,SP_X,SP_y,SP_Z,RD_X,RD_Y,RD_Z,(T60+CRR)] /5) + 1
Generete RIRs using trained model
Download the trained model using this command
source download_generate.sh
Create normalized embeddings list in pickle format. You can run following command to generate an example embedding list
python3 example1.py
Run the following command inside code_new to generate RIRs corresponding to the normalized embeddings list. You can find generated RIRs inside code_new/Generated_RIRs
python3 main.py --cfg cfg/RIR_eval.yml --gpu 0
Range
Our trained NN-DAS is capable of generating RIRs with the following range accurately.
Room Dimension X --> 8m to 11m
Room Dimesnion Y --> 6m to 8m
Room Dimension Z --> 2.5m to 3.5m
Listener Position --> Any position within the room
Speaker Position --> Any position within the room
Reverberation time --> 0.2s to 0.7s
Training the Model
Run the following command to download the training dataset we created using a Diffuse Acoustic Simulator. You also can train the model using your dataset.
source download_data.sh
Run the following command to train the model. You can pass what GPUs to be used for training as an input argument. In this example, I am using 2 GPUs.
python3 main.py --cfg cfg/RIR_s1.yml --gpu 0,1
Related Works
- IR-GAN: Room Impulse Response Generator for Far-field Speech Recognition (INTERSPEECH2021)
- TS-RIR: Translated synthetic room impulse responses for speech augmentation (IEEE ASRU 2021)
Citations
If you use our FAST-RIR for your research, please consider citing
@article{ratnarajah2021fast,
title={FAST-RIR: Fast neural diffuse room impulse response generator},
author={Ratnarajah, Anton and Zhang, Shi-Xiong and Yu, Meng and Tang, Zhenyu and Manocha, Dinesh and Yu, Dong},
journal={arXiv preprint arXiv:2110.04057},
year={2021}
}
Our work is inspired by
@inproceedings{han2017stackgan,
Author = {Han Zhang and Tao Xu and Hongsheng Li and Shaoting Zhang and Xiaogang Wang and Xiaolei Huang and Dimitris Metaxas},
Title = {StackGAN: Text to Photo-realistic Image Synthesis with Stacked Generative Adversarial Networks},
Year = {2017},
booktitle = {{ICCV}},
}
If you use our training dataset generated using Diffuse Acoustic Simulator in your research, please consider citing
@inproceedings{9052932,
author={Z. {Tang} and L. {Chen} and B. {Wu} and D. {Yu} and D. {Manocha}},
booktitle={ICASSP 2020 - 2020 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)},
title={Improving Reverberant Speech Training Using Diffuse Acoustic Simulation},
year={2020},
volume={},
number={},
pages={6969-6973},
}
170 Dec 28, 2022
33 Dec 14, 2022
28 Aug 03, 2022
35 Dec 16, 2022
125 Dec 24, 2022
78 Dec 18, 2022
14 Nov 28, 2022
42 Dec 02, 2022
4 Jan 04, 2022
5 Oct 30, 2022
100 Dec 26, 2022
118 Dec 12, 2022
643 Dec 30, 2022
9 Dec 18, 2022
134 Dec 27, 2022
575 Dec 06, 2022
22 May 28, 2022
12 Oct 18, 2022
1.3k Jan 07, 2023
1 Jan 23, 2022