项目作者: BakingBrains

项目描述 :
Sound classification on Urban Sound Dataset
高级语言: Python
项目地址: git://github.com/BakingBrains/Sound_Classification.git
创建时间: 2021-07-13T09:12:57Z
项目社区:https://github.com/BakingBrains/Sound_Classification

开源协议:

下载


Sound_Classification

Sound classification on Urban Sound Dataset

Download the urban sound dataset here

  • Extract the dataset on to a folder
  • Install pytorch (gpu verison if you have gpu)
  • Install the requirements

    1. pip install -r requirements.txt
  • Run the dataset.py file (change the path to the dataset)

    1. python dataset.py

    This will give the number of items in the dataset

  • Then run modelcnn.py to know about the structure of the CNN model.

  • Then start the training. (change the path to the dataset)

    1. python train.py
  • After the training, the model will be saved in a specified folder. Use the saved model for inferencing.

    Run

    1. python inference.py

Reference: https://github.com/musikalkemist/pytorchforaudio