class_docstr: A-to-B datasets [ZPI+17]_
  
  A general class for image-to-image translation dataset. Currently
  supported are:
  
  * Horse <-> Zebra
  * Apple <-> Orange
  * Summer <-> Winter
  
  .. [ZPI+17] Zhu, J., Park, T., Isola, P., & Efros, A.A. (2017).
      Unpaired Image-to-Image Translation Using Cycle-Consistent
      Adversarial Networks. 2017 IEEE International Conference on
      Computer Vision (ICCV), 2242-2251.
class_docstr_tok: ["A-to-B","datasets","[","ZPI+17","]","_","A","general","class","for","image-to-image","translation","dataset",".","Currently","supported","are",":","*","Horse","<","-",">","Zebra","*","Apple","<","-",">","Orange","*","Summer","<","-",">","Winter","..","[","ZPI+17","]","Zhu",",","J.",",","Park",",","T.",",","Isola",",","P.",",","&","Efros",",","A.A.","(","2017",")",".","Unpaired","Image-to-Image","Translation","Using","Cycle-Consistent","Adversarial","Networks",".","2017","IEEE","International","Conference","on","Computer","Vision","(","ICCV",")",",","2242-2251","."]
docstr_tok: []
code: def __init__(self, data_path, **kwargs):
      """ """
      ds_name = pathlib.Path(data_path).parts[-1]
      valid_names = [
          "horse2zebra",
          "apple2orange",
          "summer2winter_yosemite",
      ]
      assert (
          ds_name in valid_names
      ), f"path must end in one of {valid_names}, not {ds_name}"
      ds_kwargs = {
          "num_classes": 2,
          "mean": ch.tensor([0.5, 0.5, 0.5]),
          "custom_class": None,
          "std": ch.tensor([0.5, 0.5, 0.5]),
          "transform_train": da.TRAIN_TRANSFORMS_IMAGENET,
          "label_mapping": None,
          "transform_test": da.TEST_TRANSFORMS_IMAGENET,
      }
      ds_kwargs = self.override_args(ds_kwargs, kwargs)
      super(A2B, self).__init__(ds_name, data_path, **ds_kwargs)
  
imports: import pathlib
  import torch
  import torch.utils.data
  from .None import imagenet_models, cifar_models
  from torchvision import transforms, datasets
  from .tools import constants
  from .tools import openimgs_helpers
  from .None import data_augmentation
  from .None import loaders
  from .tools.helpers import get_label_mapping
label_desc: Use image node_id 1 to create a new A2B object from inherited base classes: DataSet with example: obj = A2B(data_path)