Classification of Urban Sound Audio Dataset using LSTM-based model.
- pytorch==1.0.1
- scipy==1.2.0
- torchvision==0.2.1
- pandas==0.24.1
- numpy==1.14.3
- torchaudio==0.2
- librosa==0.6.3
- pydub==0.23.1
-
Create a folder named data/test in the current directory which will contain all the '.wav' files that are to be tested.
-
Download 'bestModel.pt' from this Link and place in the current directory.
-
Run the following commands:
python preprocess.py
python eval.py
- A csv file named 'test_predictions.csv' will be generated in the current directory containing all the test files along with their corresponding predicted labels.
- Anish Madan
- Sarthak Bhagat
- Shagun Uppal