File size: 829 Bytes
e7824d0 |
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 |
#-*- coding: utf-8 -*-
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']
|