clean/video/fakestormer/models/__init__.py

# -*- 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",
]