Source code for chimeranet.datasets.esc50

import io
import zipfile

from .base import Dataset, load

def major_categories():
    return ['animals', 'natural', 'human', 'interior', 'exterior']

def categories_of(major_categories):
    categories = {
        'animals': ['dog', 'rooster', 'pig', 'cow', 'frog', 'cat', 'hen', 'insects', 'sheep', 'crow'],
        'natural': ['rain', 'sea_waves', 'crackling_fire', 'crickets', 'chirping_birds', 'water_drops', 'wind', 'pouring_water', 'toilet_flush', 'thunderstorm'],
        'human': ['crying_baby', 'sneezing', 'clapping', 'breathing', 'coughing', 'footsteps', 'laughing', 'brushing_teeth', 'snoring', 'drinking_sipping'],
        'interior': ['door_wood_knock', 'mouse_click', 'keyboard_typing', 'door_wood_creaks', 'can_opening', 'washing_machine', 'vacuum_cleaner', 'clock_alarm', 'clock_tick', 'glass_breaking'],
        'exterior': ['helicopter', 'chainsaw', 'siren', 'car_horn', 'engine', 'train', 'church_bells', 'airplane', 'fireworks', 'hand_saw']
    }
    if type(major_categories) == str:
        major_categories = [major_categories]
    return sum((categories[c] for c in major_categories), [])

def all_categories():
    return categories_of(major_categories())

class ESC50(Dataset):
    def __init__(self, path, category_list, fold=(1, 2, 3, 4)):
        name_col_idx, fold_col_idx, category_col_idx = 0, 1, 3
        self.path = path
        zf = zipfile.ZipFile(path)
        rows = zf.read('ESC-50-master/meta/esc50.csv')\
            .decode().splitlines()[1:]
        rows = [row.split(',') for row in rows]
        check_category = lambda r:\
            any(r[category_col_idx] == cat for cat in category_list)
        check_fold = lambda r: int(r[fold_col_idx]) in fold
        self.name_list = [
            row[name_col_idx] for row in rows
            if check_category(row) and check_fold(row)
        ]
        zf.close()
    def load(self, index, sr, offset=0., duration=None):
        zf = zipfile.ZipFile(self.path)
        name = 'ESC-50-master/audio/'+self.name_list[index]
        y = load(io.BytesIO(zf.read(name)), sr, offset, duration)
        zf.close()
        return y
    def __len__(self):
        return len(self.name_list)

    @staticmethod
    def major_categories():
        return major_categories()
    @staticmethod
    def categories_of(major_categories):
        return categories_of(major_categories)
    @staticmethod
    def all_categories():
        return all_categories()