code: def test_image_detector_data_from_folders(tmpdir):
      predict_folder = _create_synth_folders_dataset(tmpdir)
      datamodule = KeypointDetectionData.from_folders(
          predict_folder=predict_folder,
          batch_size=1,
          transform_kwargs={"image_size": (128, 128)},
      )
      data = next(iter(datamodule.predict_dataloader()))
      sample = data[0]
      assert sample[DataKeys.INPUT].shape == (128, 128, 3)
  
imports: import pytest
  from flash.core.data.io.input import DataKeys
  from flash.core.utilities.imports import _TOPIC_IMAGE_AVAILABLE
  from flash.image.keypoint_detection import KeypointDetectionData
  from tests.image.detection.test_data import _create_synth_files_dataset, _create_synth_folders_dataset
label_desc: Use image node_id 2 for calling a global function with example usage: test_image_detector_data_from_folders(tmpdir) without return types