13
Tue Feb 14 2023 13:52:27 GMT+0000 (Coordinated Universal Time)
Saved by
@cvatai
>>> 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])
}
)
content_copyCOPY
Comments