>>> train_set = TaskVisionDataset(client, TRAIN_TASK_ID,
  target_transform=ExtractBoundingBoxes(
    include_shape_types=['rectangle']))
INFO - Fetching task 77708...
INFO - Loaded data metadata from cache
INFO - Downloading chunks...
INFO - All chunks downloaded
INFO - Loaded annotations from cache
>>> train_set[2]
(
  <PIL.JpegImagePlugin.JpegImageFile image mode=RGB size=500x330 at 0x7F98C414B7F0>,
  {
    'boxes': tensor([
      [170.1163, 158.9656, 349.4313, 329.2396],
      [255.2533,  59.5135, 458.6779, 256.9108],
      [117.3765, 115.2670, 240.9382, 253.8971]
    ]),
    'labels': tensor([4, 4, 4])
  }
)