CocoDataset#

class CocoDataset[source]#

Bases: Dataset

Read Object Detection data in the COCO format.

__init__(img_dir: str, annotation_uri: str)[source]#

Constructor.

Parameters
  • img_dir (str) – Directory containing the images. Image filenames must match the image IDs in the annotations file.

  • annotation_uri (str) – URI to a JSON file containing annotations in the COCO format.

Methods

__init__(img_dir, annotation_uri)

Constructor.

__contains__(key: str) bool#
Parameters

key (str) –

Return type

bool

__getitem__(ind: int) Tuple[ndarray, Tuple[ndarray, ndarray, str]][source]#
Parameters

ind (int) –

Return type

Tuple[ndarray, Tuple[ndarray, ndarray, str]]

__init__(img_dir: str, annotation_uri: str)[source]#

Constructor.

Parameters
  • img_dir (str) – Directory containing the images. Image filenames must match the image IDs in the annotations file.

  • annotation_uri (str) – URI to a JSON file containing annotations in the COCO format.

static __new__(cls, *args: Any, **kwargs: Any) Any#
Parameters
  • args (Any) –

  • kwargs (Any) –

Return type

Any