tagCreation.py

import glob
import csv
import os
from csv_shuffler import csv_shuffler


kicksPath = r"C:\Users\suraj\OneDrive\Desktop\ML_Samples\Drums\Kick"
snaresPath = r"C:\Users\suraj\OneDrive\Desktop\ML_Samples\Drums\Snare"
openHihatsPath = r"C:\Users\suraj\OneDrive\Desktop\ML_Samples\Drums\Open Hi-Hat"
closedHihatsPath = r"C:\Users\suraj\OneDrive\Desktop\ML_Samples\Drums\Closed Hi-Hat"

with open('drums.csv', mode='w', newline='') as csvfile:
    fieldnames = ['filename', 'type', 'classid', 'filepath']
    writer = csv.DictWriter(csvfile, fieldnames = fieldnames)
    writer.writeheader()

    for dirpath, dirname, filename in os.walk(kicksPath):
        for i in filename:
            path = os.path.join(dirpath, i)
            writer.writerow({"filename": i, "type":"Kick", "classid": 0, 'filepath': path})

    for dirpath, dirname, filename in os.walk(snaresPath):
        for i in filename:
            path = os.path.join(dirpath, i)
            writer.writerow({"filename": i, "type":"Snare", "classid": 1, 'filepath': path})

    for dirpath, dirname, filename in os.walk(closedHihatsPath):
        for i in filename:
            path = os.path.join(dirpath, i)
            writer.writerow({"filename": i, "type": "ClosedHihat", "classid": 2, 'filepath': path})

    for dirpath, dirname, filename in os.walk(openHihatsPath):
        for i in filename:
            path = os.path.join(dirpath, i)
            writer.writerow({"filename": i, "type": "OpenHihat", "classid": 3, 'filepath': path})


shuffler = csv_shuffler.ShuffleCSV(input_file="drums.csv",header=True, batch_size=3000)
shuffler.shuffle_csv()