Skip to content

Matheuao/ERANNs

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

15 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

ERANNs

An implementation of ERANNs (Efficient Residual Audio Neural Networks for Audio Pattern Recognition), built using the Keras API from TensorFlow.

This architecture is based on the article "ERANNs: Efficient residual audio neural networks for audio pattern recognition", published in Pattern Recognition Letters.

I sincerely thank and acknowledge the authors of the original work β€” Sergey Verbitskiy, Vladimir Berikov, and Viacheslav Vyshegorodtsev β€” for their valuable contribution to the field. This implementation closely follows the architectural details presented in their publication.


πŸ“¦ Installation

Clone this repository and install the package locally using pip:

git clone https://github.com/Matheuao/ERANNs.git
cd ERANNS
pip install .

Requirements: Python β‰₯ 3.6, TensorFlow β‰₯ 2.4.0

If you're using a platform like Kaggle Notebooks, you can also install directly within a cell:

!git clone https://github.com/Matheuao/ERANNs.git
!pip install ./ERANNS

πŸš€ Usage

Once installed, you can import and instantiate the ERANNs model as follows:

from eranns import ERANNs

# Create an ERANNs model with sm=2, W=4, and 527 output classes
model = ERANNs(sm=2, W=4, T0=128, N=527)
model.summary()

πŸ“š Parameters

The ERANNs() function accepts the following parameters:

Parameter Description
sm Stride mode that controls temporal/frequency resolution
W Base width multiplier for the number of filters in each stage
T0 Temporal input dimension (default: 128)
N Number of output classes (default: 527)

πŸ“ Project Structure

ERANNs/
β”œβ”€β”€ eranns/
β”‚   β”œβ”€β”€ __init__.py              # Main interface
β”‚   β”œβ”€β”€ layers.py                # ARB block implementation
β”‚   └── model.py                 # ERANNs architecture definition
β”œβ”€β”€β”€β”œβ”€β”€ data/
|       β”œβ”€β”€__init__.py           # Main interface
|       β”œβ”€β”€esc_50.y              # Train dataset pipeline
|       └──esc_50_augmented.py   # Train dataset pipeline with data augmentation(still in development)
β”œβ”€β”€ setup.py                     # Package setup script
β”œβ”€β”€ requirements.txt             # Dependencies
└── README.md                    # This file

πŸ“œ License

This project is licensed under the MIT License.


🀝 Contributing

Contributions are welcome! If you have suggestions or encounter issues, feel free to open a GitHub Issue or submit a pull request.

About

a implementation of ERANNs (Efficient Residual Audio Neural Networks for Audio Pattern Recognition)

Resources

Stars

Watchers

Forks

Releases

Packages

Contributors

Languages