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: