def __init__(self, shape, encoding_format='png', ffmpeg_extra_args=()): """Initializes the connector. Args: shape: tuple of ints, the shape of the video (num_frames, height, width, channels), where channels is 1 or 3. encoding_format: The video is stored as a sequence of encoded images. You can use any encoding format supported by image_feature.Feature. ffmpeg_extra_args: A sequence of additional args to be passed to the ffmpeg binary. Specifically, ffmpeg will be called as: `` ffmpeg -i <input_file> <ffmpeg_extra_args> %010d.<encoding_format> `` Raises: ValueError: If the shape is invalid """ shape = tuple(shape) if len(shape) != 4: raise ValueError('Video shape should be of rank 4') self._encoding_format = encoding_format self._extra_ffmpeg_args = list(ffmpeg_extra_args or []) super(Video, self).__init__( image_feature.Image(shape=shape[1:], encoding_format=encoding_format), length=shape[0], )
def __init__( self, shape: Sequence[Optional[int]], encoding_format: str = 'png', ffmpeg_extra_args: Sequence[str] = (), use_colormap: bool = False, dtype=tf.uint8, doc: feature_lib.DocArg = None, ): """Initializes the connector. Args: shape: tuple of ints, the shape of the video (num_frames, height, width, channels), where channels is 1 or 3. encoding_format: The video is stored as a sequence of encoded images. You can use any encoding format supported by image_feature.Feature. ffmpeg_extra_args: A sequence of additional args to be passed to the ffmpeg binary. Specifically, ffmpeg will be called as: `` ffmpeg -i <input_file> <ffmpeg_extra_args> %010d.<encoding_format> `` use_colormap: Forwarded to `tfds.features.Image`. If `True`, `tfds.as_dataframe` will display each value in the image with a different color. dtype: tf.uint16 or tf.uint8 (default). tf.uint16 can be used only with png encoding_format doc: Documentation of this feature (e.g. description). Raises: ValueError: If the shape is invalid """ shape = tuple(shape) if len(shape) != 4: raise ValueError('Video shape should be of rank 4') self._encoding_format = encoding_format self._extra_ffmpeg_args = list(ffmpeg_extra_args or []) super(Video, self).__init__( image_feature.Image( shape=shape[1:], dtype=dtype, encoding_format=encoding_format, use_colormap=use_colormap, ), length=shape[0], )
def __init__(self, shape): """Construct the connector. Args: shape: tuple of ints, the shape of the video (num_frames, height, width, channels), where channels is 1 or 3. Raises: ValueError: If the shape is invalid """ shape = tuple(shape) if len(shape) != 4: raise ValueError('Video shape should be of rank 4') if shape.count(None) > 1: raise ValueError('Video shape cannot have more than 1 unknown dim') super(Video, self).__init__( image_feature.Image(shape=shape[1:], encoding_format='png'), length=shape[0], )