CNN that classifies different genres of music
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Failed to load latest commit information.

WyGen: music genre classifier

Convolutional Neural Network that classifies different genres of music

Based on the AlexNet architecture, the network consists of 3 convolutional layers, and 2 fully-connected layers, classifying music into 10 different genres (Blues, Classical, Country, Disco, Hip-Hop, Jazz, Metal, Pop, Rock, and Reggae). With an accuracy of 63.7% it is not state-of-the-art(6 genres:accuracy of 84%), but it performs well above the 10% chance.

Future Changes

  • Adding an LSTM (or another derivative recurrent neural net) layer that helps capture the sequential nature of music.
  • Add 1-2 more convolutional layers to capture even lower level features.
  • Train and test on more data.