class_docstr: Create image or video batch.
class_docstr_tok: ["Create","image","or","video","batch","."]
code: def __init__(
      self, channels_first, channels, image_data=True, batch_size=2
  ):
      self.channels_first = channels_first
      self.channels = (channels,)
      self.batch_size = batch_size
      self.image_data = image_data
  
file_path: adversarial-robustness-toolbox/tests/defences/preprocessor/test_jpeg_compression.py
imports: from __future__ import absolute_import, division, print_function, unicode_literals
  import logging
  import numpy
  import pytest
  from numpy.testing import assert_array_equal
  from art.config import ART_NUMPY_DTYPE
  from art.defences.preprocessor import JpegCompression
  from tests.utils import ARTTestException
label_desc: Use image node_id 1 to create a new DataGenerator object  with example: obj = DataGenerator(channels_first, channels, image_data, batch_size)