Back to top
class composer.datasets.COCODetection(img_folder, annotate_file, transforms=None)[source]
PyTorch Dataset for the COCO dataset.
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
transforms (Module) – transformations to apply to the image.