>>> 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]) } )