COCODetection

class composer.datasets.COCODetection(img_folder, annotate_file, transform=None)

PyTorch Dataset for the COCO dataset.

Parameters

• img_folder (str) – the path to the COCO folder.

• annotate_file (str) – path to a file that contains image id, annotations (e.g., bounding boxes and object classes) etc.

• transform (Module) – transformations to apply to the image.