Audio Feature Extraction Models

• Key Detection: Classifies the musical key of an audio track into relative key quality (Major/Minor).

Model Details

Tempo Model

• Model Type: Custom CNN architecture for tempo classification.
• Input: Audio segments converted to Mel spectrograms followed by autocorrelation.
• Output: Predicts Beats Per Minute (BPM) in a range from [85, 170].

Key Detection Models

• Key Class Model: Classifies into 12 relative key classes.
• Key Quality Model: Determines if the key is Major or Minor.
• Input: Audio segments converted to Mel spectrograms.
• Output:
  • Key Class: One of 12 key signatures.
  • Key Quality: Binary classification (0 for Major, 1 for Minor).

Usage

Prerequisites

• Python 3.7+
• PyTorch
• torchaudio
• transformers

Loading Models