PyTorch implementation of Hourglass-shaped Convolutional Recurrent Network (HCRN) described in our paper "Towards Modeling Auditory Restoration in Noisy Environments"
Pytorch＝0.4.1
resampy
soundfile
pysepq
librosa
pystoi
lera
Use the scripts in preprocess to process the WSJ, ESC-50 and Audioset datasets. Please modify the dataset paths in the scripts.
For training: python3 Train.py
For testing: pytho3 Test.py
Please modify specify processed dataset paths in config.yaml.
If you have any questions, please feel free to contact me at [email protected]
@article{huang2021Towards, title={Towards Modeling Auditory Restoration in Noisy Environments}, author={Huang, Yating and Hao, Yunzhe and Xu, Jiaming and Xu, Bo} journal={Proc. IJCNN 2021} year={2021} }