code: def __init__(self, data_path):
      files = self.get_files_list(data_path)
      self.data_path = data_path
      self.classes_no = len(files)
      self.files_map = {}
  
      counter = 0
      for class_no, (_, samples) in enumerate(files):
          for sample in samples:
              self.files_map[counter] = (sample, class_no)
              counter += 1
      self.size = counter
  
imports: import argparse
  import cv2
  import numpy
  import nvidia.dali
  import nvidia.dali.types
  import os
label_desc: Use image node_id 1 to create a new FilesDataSet object  with example: obj = FilesDataSet(data_path)