mirror of https://github.com/JDAI-CV/fast-reid.git
23 lines
416 B
Python
23 lines
416 B
Python
# encoding: utf-8
|
|
"""
|
|
@author: sherlock
|
|
@contact: sherlockliao01@gmail.com
|
|
"""
|
|
|
|
from . import losses
|
|
from .backbones import (
|
|
BACKBONE_REGISTRY,
|
|
build_resnet_backbone,
|
|
build_backbone,
|
|
)
|
|
from .heads import (
|
|
REID_HEADS_REGISTRY,
|
|
build_heads,
|
|
EmbeddingHead,
|
|
)
|
|
from .meta_arch import (
|
|
build_model,
|
|
META_ARCH_REGISTRY,
|
|
)
|
|
|
|
__all__ = [k for k in globals().keys() if not k.startswith("_")] |