mattricesound commited on
Commit
abb9ffa
1 Parent(s): f07a3b6

Sort dataset so that train/target lines up

Browse files
Files changed (3) hide show
  1. datasets.py +4 -2
  2. download_egfx.sh +1 -1
  3. 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 += list(self.root.glob(f"Clean/{pickup.name}/**/*.wav"))
 
 
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