clip amplitude for audio. #79

lanpa · Feb 17, 2018 · 3deaf0d · 3deaf0d
1 parent 0d83dd1
commit 3deaf0d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/tensorboardX/summary.py b/tensorboardX/summary.py
@@ -178,6 +178,9 @@ def make_image(tensor):
 def audio(tag, tensor, sample_rate=44100):
     tensor = makenp(tensor)
     tensor = tensor.squeeze()
+    if abs(tensor).max()>1:
+        print('warning: audio amplitude out of range, auto clipped.')
+        tensor = tensor.clip(-1, 1)
     assert(tensor.ndim == 1), 'input tensor should be 1 dimensional.'
 
     tensor_list = [int(32767.0 * x) for x in tensor]

diff --git a/tensorboardX/writer.py b/tensorboardX/writer.py
@@ -347,7 +347,7 @@ def add_audio(self, tag, snd_tensor, global_step=None, sample_rate=44100):
             sample_rate (int): sample rate in Hz
 
         Shape:
-            snd_tensor: :math:`(1, L)`. The values should between [-1, 1].
+            snd_tensor: :math:`(1, L)`. The values should lie between [-1, 1].
         """
         self.file_writer.add_summary(audio(tag, snd_tensor, sample_rate=sample_rate), global_step)