FIX: include num_mel_bins in hashdict

NINAnor · Jan 11, 2024 · ff311e7 · ff311e7
1 parent d305512
commit ff311e7
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 3 deletions.
diff --git a/data_utils/DCASEfewshot.py b/data_utils/DCASEfewshot.py
@@ -261,7 +261,8 @@ def preprocess_df(df):
         "frame_length": frame_length,
         "tensor_length": tensor_length,
         "set_type": set_type,
-        "overlap": overlap
+        "overlap": overlap,
+        "num_mel_bins": num_mel_bins
     }
     if resample:
         my_hash_dict["target_fs"] = target_fs

diff --git a/datamodules/DCASEDataModule.py b/datamodules/DCASEDataModule.py
@@ -102,6 +102,7 @@ def __init__(
         n_way: int = 5,
         n_subsample: int = 1,
         overlap: float = 0.5,
+        num_mel_bins: int = 128,
 
         **kwargs,
     ):
@@ -121,6 +122,7 @@ def __init__(
         self.n_way = n_way
         self.n_subsample = n_subsample
         self.overlap = overlap
+        self.num_mel_bins = num_mel_bins
         self.setup()
 
     def setup(self, stage=None):
@@ -132,7 +134,8 @@ def setup(self, stage=None):
             "frame_length": self.frame_length,
             "tensor_length": self.tensor_length,
             "set_type": self.set_type,
-            "overlap": self.overlap
+            "overlap": self.overlap,
+            "num_mel_bins": self.num_mel_bins,
         }
         if self.resample:
             my_hash_dict["target_fs"] = self.target_fs

diff --git a/datamodules/TestDCASEDataModule.py b/datamodules/TestDCASEDataModule.py
@@ -107,6 +107,7 @@ def __init__(
         n_query: int = 10,
         n_way: int = 2,
         n_subsample: int = 1,
+        num_mel_bins: int = 128,
         **kwargs
     ):
         super().__init__(**kwargs)
@@ -118,6 +119,7 @@ def __init__(
         self.n_query = n_query
         self.n_way = n_way
         self.n_subsample = n_subsample
+        self.num_mel_bins = num_mel_bins
         self.setup()
 
     def setup(self, stage=None):

diff --git a/evaluate/evaluateDCASE.py b/evaluate/evaluateDCASE.py
@@ -592,7 +592,8 @@ def write_wav(
         "frame_length": cfg["data"]["frame_length"],
         "tensor_length": cfg["data"]["tensor_length"],
         "set_type": cfg["data"]["set_type"],
-        "overlap": cfg["data"]["overlap"]
+        "overlap": cfg["data"]["overlap"],
+        "num_mel_bins": cfg["data"]["num_mel_bins"]
     }
     if cfg["data"]["resample"]:
         my_hash_dict["target_fs"] = cfg["data"]["target_fs"]