class_docstr: `ARKitScenes Dataset class.
  
  Args:
      root (string): Root directory of dataset where directory
          exists or will be saved to if download is set to True.
      transform (callable, optional): A function that takes in a sample (dict)
          and returns a transformed version.
      download (bool, optional): If true, downloads the dataset from the internet and
          puts on the root directory. If dataset is already downloaded, it is not
          downloaded again.
docstr: Args:
      index (int): Index
  
  Returns:
      tuple: (identifier, color, highres_depth, lowres_depth).
docstr_tok: ["Args",":","index","(","int",")",":","Index","Returns",":","tuple",":","(","identifier",",","color",",","highres_depth",",","lowres_depth",")","."]
code: def __getitem__(self, index: int):
      """
      Args:
          index (int): Index
  
      Returns:
          tuple: (identifier, color, highres_depth, lowres_depth).
      """
      video_id, sample_id, direction = self.samples[index]
      sample = dict()
      sample[dataset_keys.IDENTIFIER] = str(sample_id)
  
      rgb_file = os.path.join(
          self.dataset_folder, video_id, WIDE, sample_id
      )
      depth_file = os.path.join(
          self.dataset_folder, video_id, "highres_depth", sample_id
      )
      apple_depth_file = os.path.join(
          self.dataset_folder, video_id, "lowres_depth", sample_id
      )
  
      sample[dataset_keys.COLOR_IMG] = self.load_image(
          rgb_file, self.high_res, False, direction
      )
      sample[dataset_keys.HIGH_RES_DEPTH_IMG] = self.load_image(
          depth_file, self.high_res, True, direction
      )
      sample[dataset_keys.LOW_RES_DEPTH_IMG] = self.load_image(
          apple_depth_file, self.low_res, True, direction
      )
  
      if self.transform is not None:
          sample = self.transform(sample)
      return sample
  
file_path: ARKitScenes/depth_upsampling/dataset.py
imports: import os
  from glob import glob
  from typing import Callable, Optional
  import cv2
  import numpy
  import pandas
  from torch.utils.data import Dataset
  import dataset_keys
  from data_utils import image_hwc_to_chw, expand_channel_dim
label_desc: Use image node_id 4 for calling the ARKitScenesDataset obj's underlying member method code with example usage: obj.__getitem__(index) and returns: sample