| from pathlib import Path | |
| def midi_explorer(dataset, path): # path(str)->paths(list) | |
| if dataset == "lakh": | |
| return list(Path(path).glob("*.mid")) | |
| if dataset == "lakh_matched": | |
| return list(Path(path).rglob("*.mid")) | |
| if dataset == "guitarset": | |
| return list(Path(path).glob("*.jams")) | |
| if dataset == "maestro": | |
| files = [] | |
| with open(path.split(",")[0] + "/maestro-v3.0.0.csv", "r", encoding="utf-8") as file: | |
| reader = csv.reader(file) | |
| next(reader) | |
| for row in reader: | |
| if row[2] == path.split(",")[1]: | |
| files.append(path.split(",")[0] + "/" + row[4]) | |
| return files | |
| if dataset == "slakh": | |
| return list(Path(path).glob("**/MIDI/*.mid")) | |
| if dataset == "molina": | |
| return list(Path(path).glob("*.GroundTruth.txt")) | |
| if dataset == "phenicx": | |
| return list(Path(path).glob("*.mid")) | |
| if dataset == "urmp": | |
| return list(Path(path).rglob("Notes*.txt")) | |
| if dataset == "gaps": | |
| return list(Path(path).glob("*.mid")) | |
| if dataset == "other": | |
| return list(Path(path).glob("*.mid")) | |
| if dataset == "maps": | |
| return list(Path(path).glob("*.txt")) | |