coco#
COCO (Common Objects in Context) dataset.
COCO is a large-scale object detection, segmentation, and captioning dataset. Please refer to the COCO dataset for more details.
Classes
PyTorch Dataset for the COCO dataset. |
|
Implementation of the COCO dataset using StreamingDataset. |
- class composer.datasets.coco.COCODetection(img_folder, annotate_file, transform=None)[source]#
Bases:
torch.utils.data.dataset.Dataset
PyTorch Dataset for the COCO dataset.
- class composer.datasets.coco.StreamingCOCO(remote, local, split, shuffle, batch_size=None)[source]#
Bases:
composer.datasets.streaming.dataset.StreamingDataset
,torchvision.datasets.vision.VisionDataset
Implementation of the COCO dataset using StreamingDataset.
- Parameters
remote (str) โ Remote directory (S3 or local filesystem) where dataset is stored.
local (str) โ Local filesystem directory where dataset is cached during operation.
split (str) โ The dataset split to use, either โtrainโ or โvalโ.
shuffle (bool) โ Whether to shuffle the samples in this dataset.
batch_size (Optional[int]) โ Hint the batch_size that will be used on each deviceโs DataLoader. Default:
None
.