D-FINE/src/data/dataset/coco_eval.py

"""
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
COCO evaluator that works in distributed mode.
Mostly copy-paste from https://github.com/pytorch/vision/blob/edfd5a7/references/detection/coco_eval.py
The difference is that there is less copy-pasting from pycocotools
in the end of the file, as python3 can suppress prints with contextlib

# MiXaiLL76 replacing pycocotools with faster-coco-eval for better performance and support.
"""

from faster_coco_eval.utils.pytorch import FasterCocoEvaluator

from ...core import register

__all__ = [
    "CocoEvaluator",
]


@register()
class CocoEvaluator(FasterCocoEvaluator):
    pass