Datasets

Coco Format Dataset

class relai.datasets.coco_format.CocoFormatDataset(dataset_path: str | Path, annFile: str | Path)

Bases: RELAIDataset

Load any object detection dataset that follows COCO data format (https://cocodataset.org/#format-data). It requires the COCO API (pycocotools) to be installed (https://github.com/cocodataset/cocoapi/tree/master).

• Parameters:
  • dataset_path (Union *[*str , Path ]) – Root directory where images are stored.
  • annFile (Union *[*str , Path ]) – Path to the corresponding json annotation file.