Add read_wav and list_wav_info support (#406)

* Add read_wav and list_wav_info support This PR adds read_wav and list_wav_info support so that it is possible to read wav file into a tensor. Since WAV file is splittable, read_wav could take a start and count parameter so that only a slice of the wav file is read. This is also part of the effort to rework on Dataset to move to primitive ops. See See 382 and 366 for related discussions. Signed-off-by: Yong Tang <yong.tang.github@outlook.com> * Rename start, count to start, stop Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
tensorflow · Aug 4, 2019 · 8af0a01 · 8af0a01
1 parent 7b05cd4
commit 8af0a01
Show file tree

Hide file tree

Showing 10 changed files with 438 additions and 213 deletions.
diff --git a/tensorflow_io/audio/BUILD b/tensorflow_io/audio/BUILD
@@ -10,7 +10,7 @@ load(
 cc_library(
     name = "audio_ops",
     srcs = [
-        "kernels/audio_input.cc",
+        "kernels/audio_kernels.cc",
         "ops/audio_ops.cc",
     ],
     copts = tf_io_copts(),

diff --git a/tensorflow_io/audio/__init__.py b/tensorflow_io/audio/__init__.py
@@ -15,18 +15,24 @@
 """Audio Dataset.
 
 @@WAVDataset
+@@list_wav_info
+@@read_wav
 """
 
 from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
 from tensorflow_io.audio.python.ops.audio_ops import WAVDataset
+from tensorflow_io.audio.python.ops.audio_ops import list_wav_info
+from tensorflow_io.audio.python.ops.audio_ops import read_wav
 
 from tensorflow.python.util.all_util import remove_undocumented
 
 _allowed_symbols = [
     "WAVDataset",
+    "list_wav_info",
+    "read_wav",
 ]
 
 remove_undocumented(__name__)
diff --git a/tensorflow_io/audio/kernels/audio_input.cc b/tensorflow_io/audio/kernels/audio_input.cc