code: def initialize_augmentation(self):
      if not self.initialization:
          for label, data_group in list(self.data_groups.items()):
              self.input_shape[label] = data_group.get_shape()
  
          self.initialization = True
  
imports: import numpy
  from deepneuro.utilities.util import add_parameter
label_desc: Use image node_id 2 for calling the Downsample obj's underlying member method code with example usage: obj.initialize_augmentation() without return types