neodroidvision.data.detection.coco.coco.COCODataset

class neodroidvision.data.detection.coco.coco.COCODataset(data_root: Path, dataset_name: str, split: SplitEnum, img_transform: Optional[callable] = None, annotation_transform: Optional[callable] = None)[source]

Bases: ObjectDetectionDataset

COCO 2014

__init__(data_root: Path, dataset_name: str, split: SplitEnum, img_transform: Optional[callable] = None, annotation_transform: Optional[callable] = None)[source]
Parameters

  • data_root –

  • dataset_name –

  • split –

  • img_transform –

  • annotation_transform –

  • remove_empty –

Methods

__init__(data_root, dataset_name, split[, ...])

param data_root

get_annotation(index)

param index

get_img_info(index)

param index

Attributes

categories

image_dirs

predictor_shape

description

response_shape

param self

split_names

return: :rtype:

splits

get_annotation(index)[source]
Parameters

index –

Returns

Return type

get_img_info(index)[source]
Parameters

index –

Returns

Return type

property predictor_shape: Tuple[int, ...]

description

property response_shape: Tuple[int, ...]
Parameters

self –

Returns:

property split_names: Dict[SplitEnum, str]

return: :rtype: