fast-reid/fastreid/modeling/heads/__init__.py

14 lines
327 B
Python

# encoding: utf-8
"""
@author: liaoxingyu
@contact: sherlockliao01@gmail.com
"""
from .build import REID_HEADS_REGISTRY, build_heads
# import all the meta_arch, so they will be registered
from .embedding_head import EmbeddingHead
from .clas_head import ClasHead
from .pair_head import PairHead
from .pcb_head import PcbHead