Adds Y4MDataset for reading raw video files in .y4m format.

PiperOrigin-RevId: 363034108
Change-Id: I61b43fc5ce1df1015307fd87b34779e9aacdcb05

Author: Johannes Ballé

BUILD

@@ -8,6 +8,7 @@ py_library(
     srcs_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
+        "//tensorflow_compression/python/datasets:y4m_dataset",
         "//tensorflow_compression/python/distributions:deep_factorized",
         "//tensorflow_compression/python/distributions:helpers",
         "//tensorflow_compression/python/distributions:round_adapters",
@@ -40,6 +41,7 @@ py_binary(
         "tensorflow_compression/all_tests.py",
         ":tensorflow_compression",
         # The following targets are for Python unit tests.
+        "//tensorflow_compression/python/datasets:py_src",
         "//tensorflow_compression/python/distributions:py_src",
         "//tensorflow_compression/python/entropy_models:py_src",
         "//tensorflow_compression/python/layers:py_src",

tensorflow_compression/__init__.py

@@ -15,6 +15,8 @@
 """Data compression in TensorFlow."""
 
 # pylint: disable=wildcard-import
+from tensorflow_compression.python.datasets.y4m_dataset import *
+
 from tensorflow_compression.python.distributions.deep_factorized import *
 from tensorflow_compression.python.distributions.helpers import *
 from tensorflow_compression.python.distributions.round_adapters import *

tensorflow_compression/all_tests.py

@@ -20,6 +20,8 @@
 import tensorflow as tf
 
 # pylint: disable=wildcard-import
+from tensorflow_compression.python.datasets.y4m_dataset_test import *
+
 from tensorflow_compression.python.distributions.deep_factorized_test import *
 from tensorflow_compression.python.distributions.helpers_test import *
 from tensorflow_compression.python.distributions.round_adapters_test import *