File size: 462 Bytes
e1e7af0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | """
Copied from RT-DETR (https://github.com/lyuwenyu/RT-DETR)
Copyright(c) 2023 lyuwenyu. All Rights Reserved.
"""
from .common import (
get_activation,
FrozenBatchNorm2d,
freeze_batch_norm2d,
)
from .presnet import PResNet
from .test_resnet import MResNet
from .timm_model import TimmModel
from .torchvision_model import TorchVisionModel
from .csp_resnet import CSPResNet
from .csp_darknet import CSPDarkNet, CSPPAN
from .hgnetv2 import HGNetv2
|