|
|
|
|
|
from .builder import MODELS, build_model |
|
|
from .networks.arcface import ( |
|
|
SimpleClassificationDF, |
|
|
) |
|
|
from .networks.mrsa_resnet import ( |
|
|
PoseResNet, resnet_spec, Bottleneck |
|
|
) |
|
|
from .networks.pose_hrnet import ( |
|
|
PoseHighResolutionNet |
|
|
) |
|
|
from .networks.xception import ( |
|
|
Xception |
|
|
) |
|
|
from.networks.pose_efficientNet import ( |
|
|
PoseEfficientNet |
|
|
) |
|
|
from .networks.common import * |
|
|
from .utils import ( |
|
|
load_pretrained, freeze_backbone, |
|
|
load_model, save_model, unfreeze_backbone, |
|
|
preset_model, |
|
|
) |
|
|
|
|
|
|
|
|
__all__=['SimpleClassificationDF', 'PoseResNet', 'MODELS', 'build_model', |
|
|
'load_pretrained', 'freeze_backbone', 'resnet_spec', |
|
|
'load_model', 'save_model', 'unfreeze_backbone', 'Bottleneck', |
|
|
'preset_model', 'PoseHighResolutionNet', 'Xception', 'PoseEfficientNet'] |
|
|
|
