'fix': fix compute data statistics wrong when frame rate is 250 in datasets

Author: JinFoolish

ddsp/colab/colab_utils.py

@@ -183,7 +183,8 @@ def upload(sample_rate=DEFAULT_SAMPLE_RATE, normalize_db=None):
 def save_dataset_statistics(data_provider,
                             file_path=None,
                             batch_size=1,
-                            power_frame_size=256,):
+                            power_frame_size=256,
+                            power_frame_rate=250):
   """Calculate dataset stats and save in a pickle file.
 
   Calls out to postprocessing.compute_dataset_statistics.
@@ -201,7 +202,7 @@ def save_dataset_statistics(data_provider,
   """
 
   ds_stats = ddsp.training.postprocessing.compute_dataset_statistics(
-      data_provider, batch_size, power_frame_size)
+      data_provider, batch_size, power_frame_size, power_frame_rate)
 
   # Save.
   if file_path is not None:

ddsp/training/postprocessing.py

@@ -325,6 +325,10 @@ def get_stats(x, prefix='x', note_mask=None):
       mean_min = np.mean(np.min(x, axis=-1))
     else:
       max_list = []
+      if x.shape != note_mask.shape:
+        min_length = min(x.shape[1], note_mask.shape[1])
+        x = x[:, :min_length]
+        note_mask = note_mask[:, :min_length]
       for x_i, m in zip(x, note_mask):
         if np.sum(m) > 0:
           max_list.append(np.max(x_i[m]))