avalanche.evaluation.metrics.detection.get_detection_api_from_dataset
- avalanche.evaluation.metrics.detection.get_detection_api_from_dataset(dataset, supported_types: ~typing.Sequence[~typing.Tuple[str, ~typing.Union[~typing.Tuple[~typing.Type], ~typing.Type]]] = (('coco', <class 'pycocotools.coco.COCO'>), ('lvis_api', <class 'lvis.lvis.LVIS'>)), default_to_coco: bool = True, none_if_not_found=False)[source]
Adapted from: https://github.com/pytorch/vision/blob/main/references/detection/engine.py
- Parameters
dataset – The test dataset.
supported_types – The supported API types
default_to_coco – If True, if no API object can be found, the dataset will be converted to COCO.
none_if_not_found – If True, it will return None if no valid detection API object is found. Else, it will consider default_to_coco or will raise an error.
- Returns
The detection object.