# @package data
train_dataset:
  _target_: datasets.semseg.SemanticSegmentationDataset
  dataset_name: "stpls3d"
  data_dir: data/processed/stpls3d
  image_augmentations_path: conf/augmentation/albumentations_aug.yaml
  volume_augmentations_path: conf/augmentation/volumentations_aug.yaml
  label_db_filepath: data/processed/stpls3d/label_database.yaml
  color_mean_std: data/processed/stpls3d/color_mean_std.yaml
  data_percent: 1.0
  mode: ${data.train_mode}
  ignore_label: ${data.ignore_label}
  num_labels: ${data.num_labels}
  add_raw_coordinates: ${data.add_raw_coordinates}
  add_colors: ${data.add_colors}
  add_normals: ${data.add_normals}
  add_instance: ${data.add_instance}
  cache_data: ${data.cache_data}
  # different augs experiments
  instance_oversampling: 0.0
  place_around_existing: False
  point_per_cut: 0
  max_cut_region: 0
  flip_in_center: false
  noise_rate: 0
  resample_points: 0
  cropping: ${data.cropping}
  cropping_args: ${data.cropping_args}
  is_tta: false
  crop_min_size: ${data.crop_min_size}
  crop_length: ${data.crop_length}
  cropping_v1: ${data.cropping_v1}
  area: ${general.area}
  reps_per_epoch: ${general.reps_per_epoch}
  eval_inner_core: ${general.eval_inner_core}
  filter_out_classes: [0]
  label_offset: 1
  is_elastic_distortion: true
  color_drop: 0.0

validation_dataset:
  _target_: datasets.semseg.SemanticSegmentationDataset
  dataset_name: "stpls3d"
  data_dir: data/processed/stpls3d
  image_augmentations_path: null
  volume_augmentations_path: null
  label_db_filepath: data/processed/stpls3d/label_database.yaml
  color_mean_std: data/processed/stpls3d/color_mean_std.yaml
  data_percent: 1.0
  mode: ${data.validation_mode}
  ignore_label: ${data.ignore_label}
  num_labels: ${data.num_labels}
  add_raw_coordinates: ${data.add_raw_coordinates}
  add_colors: ${data.add_colors}
  add_normals: ${data.add_normals}
  add_instance: ${data.add_instance}
  cache_data: ${data.cache_data}
  cropping: false
  is_tta: false
  crop_min_size: ${data.crop_min_size}
  crop_length: ${data.crop_length}
  cropping_v1: ${data.cropping_v1}
  area: ${general.area}
  on_crops: ${general.on_crops}
  eval_inner_core: ${general.eval_inner_core}
  filter_out_classes: [0]
  label_offset: 1

test_dataset:
  _target_: datasets.semseg.SemanticSegmentationDataset
  dataset_name: "stpls3d"
  data_dir: data/processed/stpls3d
  image_augmentations_path: null
  volume_augmentations_path: null
  label_db_filepath: data/processed/stpls3d/label_database.yaml
  color_mean_std: data/processed/stpls3d/color_mean_std.yaml
  data_percent: 1.0
  mode: ${data.test_mode}
  ignore_label: ${data.ignore_label}
  num_labels: ${data.num_labels}
  add_raw_coordinates: ${data.add_raw_coordinates}
  add_colors: ${data.add_colors}
  add_normals: ${data.add_normals}
  add_instance: ${data.add_instance}
  cache_data: ${data.cache_data}
  cropping: false
  is_tta: false
  crop_min_size: ${data.crop_min_size}
  crop_length: ${data.crop_length}
  cropping_v1: ${data.cropping_v1}
  area: ${general.area}
  on_crops: ${general.on_crops}
  eval_inner_core: ${general.eval_inner_core}
  filter_out_classes: [0]
  label_offset: 1
