File size: 1,313 Bytes
9e14838 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 | # -*- coding: utf-8 -*-
from .builder import MODELS, build_model
from .networks.backbones import (
ResNet3D,
SwinTransformer,
SwinTransformer3D,
TimeViT,
ViT,
Xception,
)
from .networks.backbones.arcface import (
SimpleClassificationDF,
)
from .networks.common import *
from .networks.detectors import TopDownDetector
from .networks.heads.hm_simple_head import TopdownHeatmapSimpleHead
from .networks.mrsa_resnet import Bottleneck, PoseResNet, resnet_spec
from .networks.necks import EFPN3D
from .networks.pose_efficientNet import PoseEfficientNet
from .networks.pose_hrnet import PoseHighResolutionNet
from .utils import (
freeze_backbone,
load_model,
load_pretrained,
n_param_model,
preset_model,
save_model,
unfreeze_backbone,
)
__all__ = [
"SimpleClassificationDF",
"PoseResNet",
"MODELS",
"build_model",
"load_pretrained",
"freeze_backbone",
"resnet_spec",
"n_param_model",
"load_model",
"save_model",
"unfreeze_backbone",
"Bottleneck",
"preset_model",
"PoseHighResolutionNet",
"Xception",
"PoseEfficientNet",
"TopDownDetector",
"ViT",
"TopdownHeatmapSimpleHead",
"TimeViT",
"SwinTransformer",
"ResNet3D",
"Xception",
"SwinTransformer3D",
"EFPN3D",
]
|