Handle edge case of no notes detected.

jesseengel · Magenta Team · commit def2c6b4540c · 2022-04-20T16:01:22.000-07:00
PiperOrigin-RevId: 443217202
diff --git a/ddsp/training/postprocessing.py b/ddsp/training/postprocessing.py
@@ -309,6 +309,10 @@ def compute_dataset_statistics(data_provider,
   # Detect notes.
   mask_on, _ = detect_notes(loudness_trimmed, f0_conf_trimmed)
 
+  # If no notes detected, just default to using full signal.
+  mask_on = np.logical_or(
+      mask_on, np.logical_not(np.any(mask_on, axis=1, keepdims=True)))
+
   quantile_transform = fit_quantile_transform(loudness_trimmed, mask_on)
 
   # Pitch statistics.
diff --git a/ddsp/version.py b/ddsp/version.py
@@ -18,4 +18,4 @@
 pulling in all the dependencies in __init__.py.
 """
 
-__version__ = '3.3.5'
+__version__ = '3.3.6'
