text: from configs import transforms_config
  from configs.paths_config import dataset_paths
  
  
  DATASETS = {
  	'ffhq_encode': {
  		'transforms': transforms_config.EncodeTransforms,
  		'train_source_root': dataset_paths['ffhq'],
  		'train_target_root': dataset_paths['ffhq'],
  		'test_source_root': dataset_paths['celeba_test'],
  		'test_target_root': dataset_paths['celeba_test'],
  	},
  	'ffhq_frontalize': {
  		'transforms': transforms_config.FrontalizationTransforms,
  		'train_source_root': dataset_paths['ffhq'],
  		'train_target_root': dataset_paths['ffhq'],
  		'test_source_root': dataset_paths['celeba_test'],
  		'test_target_root': dataset_paths['celeba_test'],
  	},
  	'celebs_sketch_to_face': {
  		'transforms': transforms_config.SketchToImageTransforms,
  		'train_source_root': dataset_paths['celeba_train_sketch'],
  		'train_target_root': dataset_paths['celeba_train'],
  		'test_source_root': dataset_paths['celeba_test_sketch'],
  		'test_target_root': dataset_paths['celeba_test'],
  	},
  	'celebs_seg_to_face': {
  		'transforms': transforms_config.SegToImageTransforms,
  		'train_source_root': dataset_paths['celeba_train_segmentation'],
  		'train_target_root': dataset_paths['celeba_train'],
  		'test_source_root': dataset_paths['celeba_test_segmentation'],
  		'test_target_root': dataset_paths['celeba_test'],
  	},
  	'celebs_super_resolution': {
  		'transforms': transforms_config.SuperResTransforms,
  		'train_source_root': dataset_paths['celeba_train'],
  		'train_target_root': dataset_paths['celeba_train'],
  		'test_source_root': dataset_paths['celeba_test'],
  		'test_target_root': dataset_paths['celeba_test'],
  	},
  	'sketch_to_zookiz': {
  		'transforms': transforms_config.ZookizTransforms,
  		'train_source_root': dataset_paths['zookiz_sketch_train'],
  		'train_target_root': dataset_paths['zookiz_train'],
  		'test_source_root': dataset_paths['zookiz_test_sktech'],
  		'test_target_root': dataset_paths['zookiz_test'],
  	}
  }
  