external/det/mmdet/evaluation/functional/__init__.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.

from .bbox_overlaps import bbox_overlaps
from .cityscapes_utils import evaluateImgLists
from .class_names import (cityscapes_classes, coco_classes,
                          coco_panoptic_classes, dataset_aliases, get_classes,
                          imagenet_det_classes, imagenet_vid_classes,
                          objects365v1_classes, objects365v2_classes,
                          oid_challenge_classes, oid_v6_classes, voc_classes)
from .mean_ap import average_precision, eval_map, print_map_summary
from .panoptic_utils import (INSTANCE_OFFSET, pq_compute_multi_core,
                             pq_compute_single_core)
from .recall import (eval_recalls, plot_iou_recall, plot_num_recall,
                     print_recall_summary)
from .ytvis import YTVIS
from .ytviseval import YTVISeval

__all__ = [
    'voc_classes', 'imagenet_det_classes', 'imagenet_vid_classes',
    'coco_classes', 'cityscapes_classes', 'dataset_aliases', 'get_classes',
    'average_precision', 'eval_map', 'print_map_summary', 'eval_recalls',
    'print_recall_summary', 'plot_num_recall', 'plot_iou_recall',
    'oid_v6_classes', 'oid_challenge_classes', 'INSTANCE_OFFSET',
    'pq_compute_single_core', 'pq_compute_multi_core', 'bbox_overlaps',
    'objects365v1_classes', 'objects365v2_classes', 'coco_panoptic_classes',
    'evaluateImgLists', 'YTVIS', 'YTVISeval'
]