class_docstr: Dataset wrapper to perform AugMix or other clean/augmentation mixes
class_docstr_tok: ["Dataset","wrapper","to","perform","AugMix","or","other","clean\/augmentation","mixes"]
code: def __init__(self, dataset, num_splits=2):
      self.augmentation = None
      self.normalize = None
      self.dataset = dataset
      if self.dataset.transform is not None:
          self._set_transforms(self.dataset.transform)
      self.num_splits = num_splits
  
imports: import io
  import logging
  from typing import Optional
  import torch
  import torch.utils.data
  from PIL import Image
  from .readers import create_reader
label_desc: Use image node_id 1 to create a new AugMixDataset object from inherited base classes: torch.utils.data with example: obj = AugMixDataset(dataset, num_splits)