D-FINE/src/nn/__init__.py

"""
Copied from RT-DETR (https://github.com/lyuwenyu/RT-DETR)
Copyright(c) 2023 lyuwenyu. All Rights Reserved.
"""

from .arch import *

#
from .backbone import *
from .backbone import (
    FrozenBatchNorm2d,
    freeze_batch_norm2d,
    get_activation,
)
from .criterion import *
from .postprocessor import *