code: def sample_dataset(samples_source):
      images_dir = samples_source / "images"
      images_files = get_files(images_dir, extensions=".jpg")[-2:]
  
      images = [np.array(open_img(path)) for path in images_files]
      images = [cv2.resize(image, (128, 128)) for image in images]
  
      return Dataset.from_images(images)
  
file_path: icevision/tests/models/torchvision_models/mask_rcnn/test_predict.py
imports: import pytest
  from icevision.imports import *
  from icevision import *
  from icevision.models.torchvision import mask_rcnn
label_desc: Use image node_id 1 for calling a global function with example usage: sample_dataset(samples_source) and returns: Dataset