code: def __init__(self, local_rank, **kwargs):
      super(DataLoaderX, self).__init__(**kwargs)
      self.stream = torch.cuda.Stream(local_rank)
      self.local_rank = local_rank
  
imports: import os
  import os.path
  import queue
  import pickle
  import threading
  import logging
  import numpy
  import torch
  from torch.utils.data import DataLoader, Dataset
  from torchvision import transforms
  import cv2
  import albumentations
  from albumentations.pytorch import ToTensorV2
  from .augs import RectangleBorderAugmentation
label_desc: Use image node_id 1 to create a new DataLoaderX object from inherited base classes: DataLoader with example: obj = DataLoaderX(local_rank)