params.py

SOUNDS_DATASET_PATH = "G:\Shared drives\PFE - ING3 Mlamali\DrumClassifier - Sounds Dataset"
AIRS_PATH = "G:\Shared drives\PFE - ING3 Mlamali\DrumClassifier - Acoustic Impulse Responses"

SAMPLE_RATE = 44100

DRUM_NAMES = ['Kick',
               'Tom',
               '808',
               'Ride (Cymbal)',
               'Crash (Cymbal)',
               'Hat Closed (Cymbal)',
               'Hat Open (Cymbal)',
               'Clap',
               'Snap',
               'Snare',
               'Conga',
               'Bell',
               'Shaker',
               'Metal']

HIGH_PITCHED_DRUMS = ["Shaker", "Hat Open (Cymbal)", "Hat Closed (Cymbal)", "Crash (Cymbal)", "Ride (Cymbal)"]
LOW_PITCHED_DRUMS = ['Kick', 'Tom', '808']

CLASS_COLORS = {
    'Metal': '#6a3d9a',  # Purple
    'Conga': '#1f78b4',  # Blue
    'Kick': '#984ea3',  # Dark purple
    'Hat Open (Cymbal)': '#ff7f00',  # Orange
    'Ride (Cymbal)': '#ffff33',  # Yellow
    '808': '#4d004b',  # Dark violet
    'Snare': '#e31a1c',  # Red
    'Bell': '#f781bf',  # Pink
    'Hat Closed (Cymbal)': '#ffcc00',  # Light yellow
    'Crash (Cymbal)': '#33a02c',  # Green
    'Tom': '#a6cee3',  # Light blue
    'Clap': '#fb9a99',  # Light red
    'Snap': '#b2df8a',  # Light green
    'Shaker': '#cab2d6',  # Light purple
}

DRUM_ALIAS_3 = {
    'Kick': 'KIK',
    'Tom': 'TOM',
    '808': 'EIG',
    'Ride (Cymbal)': 'RID',
    'Crash (Cymbal)': 'CRA',
    'Hat Closed (Cymbal)': 'HCL',
    'Hat Open (Cymbal)': 'HOP',
    'Clap': 'CLP',
    'Snap': 'SNP',
    'Snare': 'SNR',
    'Conga': 'CNG',
    'Bell': 'BEL',
    'Shaker': 'SHA',
    'Metal': 'MET'
}

DRUM_ALIAS_2 = {
    'Kick': 'KK',
    'Tom': 'TM',
    '808': 'EG',
    'Ride (Cymbal)': 'RD',
    'Crash (Cymbal)': 'CR',
    'Hat Closed (Cymbal)': 'HC',
    'Hat Open (Cymbal)': 'HO',
    'Clap': 'CP',
    'Snap': 'SP',
    'Snare': 'SR',
    'Conga': 'CG',
    'Bell': 'BL',
    'Shaker': 'SH',
    'Metal': 'MT'
}