Spaces:
Sleeping
Sleeping
Commit
·
6990e4a
1
Parent(s):
78ec3fa
Fix render_root naming2
Browse files- remfx/datasets.py +3 -3
remfx/datasets.py
CHANGED
@@ -20,14 +20,14 @@ class VocalSet(Dataset):
|
|
20 |
chunk_size_in_sec: int = 3,
|
21 |
effect_types: List[torch.nn.Module] = None,
|
22 |
render_files: bool = True,
|
23 |
-
|
24 |
mode: str = "train",
|
25 |
):
|
26 |
super().__init__()
|
27 |
self.chunks = []
|
28 |
self.song_idx = []
|
29 |
self.root = Path(root)
|
30 |
-
self.
|
31 |
self.chunk_size_in_sec = chunk_size_in_sec
|
32 |
self.sample_rate = sample_rate
|
33 |
self.mode = mode
|
@@ -37,7 +37,7 @@ class VocalSet(Dataset):
|
|
37 |
self.normalize = effects.LoudnessNormalize(sample_rate, target_lufs_db=-20)
|
38 |
self.effect_types = effect_types
|
39 |
|
40 |
-
self.processed_root = self.
|
41 |
|
42 |
self.num_chunks = 0
|
43 |
print("Total files:", len(self.files))
|
|
|
20 |
chunk_size_in_sec: int = 3,
|
21 |
effect_types: List[torch.nn.Module] = None,
|
22 |
render_files: bool = True,
|
23 |
+
render_root: str = None,
|
24 |
mode: str = "train",
|
25 |
):
|
26 |
super().__init__()
|
27 |
self.chunks = []
|
28 |
self.song_idx = []
|
29 |
self.root = Path(root)
|
30 |
+
self.render_root = Path(render_root)
|
31 |
self.chunk_size_in_sec = chunk_size_in_sec
|
32 |
self.sample_rate = sample_rate
|
33 |
self.mode = mode
|
|
|
37 |
self.normalize = effects.LoudnessNormalize(sample_rate, target_lufs_db=-20)
|
38 |
self.effect_types = effect_types
|
39 |
|
40 |
+
self.processed_root = self.render_root / "processed" / self.mode
|
41 |
|
42 |
self.num_chunks = 0
|
43 |
print("Total files:", len(self.files))
|