Spaces:
Runtime error
Runtime error
Commit
·
abb9ffa
1
Parent(s):
f07a3b6
Sort dataset so that train/target lines up
Browse files- datasets.py +4 -2
- download_egfx.sh +1 -1
- egfx.ipynb +0 -0
datasets.py
CHANGED
|
@@ -31,8 +31,10 @@ class GuitarFXDataset(Dataset):
|
|
| 31 |
]
|
| 32 |
for i, effect in enumerate(effect_type):
|
| 33 |
for pickup in Path(self.root / effect).iterdir():
|
| 34 |
-
self.wet_files += list(pickup.glob("*.wav"))
|
| 35 |
-
self.dry_files +=
|
|
|
|
|
|
|
| 36 |
self.labels += [i] * len(self.wet_files)
|
| 37 |
print(
|
| 38 |
f"Found {len(self.wet_files)} wet files and {len(self.dry_files)} dry files"
|
|
|
|
| 31 |
]
|
| 32 |
for i, effect in enumerate(effect_type):
|
| 33 |
for pickup in Path(self.root / effect).iterdir():
|
| 34 |
+
self.wet_files += sorted(list(pickup.glob("*.wav")))
|
| 35 |
+
self.dry_files += sorted(
|
| 36 |
+
list(self.root.glob(f"Clean/{pickup.name}/**/*.wav"))
|
| 37 |
+
)
|
| 38 |
self.labels += [i] * len(self.wet_files)
|
| 39 |
print(
|
| 40 |
f"Found {len(self.wet_files)} wet files and {len(self.dry_files)} dry files"
|
download_egfx.sh
CHANGED
|
@@ -16,6 +16,6 @@ wget https://zenodo.org/record/7044411/files/Spring-Reverb.zip?download=1 -O Spr
|
|
| 16 |
wget https://zenodo.org/record/7044411/files/Sweep-Echo.zip?download=1 -O Sweep-Echo.zip
|
| 17 |
wget https://zenodo.org/record/7044411/files/TapeEcho.zip?download=1 -O TapeEcho.zip
|
| 18 |
wget https://zenodo.org/record/7044411/files/TubeScreamer.zip?download=1 -O TubeScreamer.zip
|
| 19 |
-
unzip \*.zip
|
| 20 |
|
| 21 |
|
|
|
|
| 16 |
wget https://zenodo.org/record/7044411/files/Sweep-Echo.zip?download=1 -O Sweep-Echo.zip
|
| 17 |
wget https://zenodo.org/record/7044411/files/TapeEcho.zip?download=1 -O TapeEcho.zip
|
| 18 |
wget https://zenodo.org/record/7044411/files/TubeScreamer.zip?download=1 -O TubeScreamer.zip
|
| 19 |
+
unzip -n \*.zip
|
| 20 |
|
| 21 |
|
egfx.ipynb
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|