mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
Summary: merge BNneckHead, LinearHead and ReductionHead into EmbeddingHead because they are highly similar and can be prepared for ClsHead
11 lines
238 B
Python
11 lines
238 B
Python
# encoding: utf-8
|
|
"""
|
|
@author: liaoxingyu
|
|
@contact: sherlockliao01@gmail.com
|
|
"""
|
|
|
|
from .build import REID_HEADS_REGISTRY, build_reid_heads
|
|
|
|
# import all the meta_arch, so they will be registered
|
|
from .embedding_head import EmbeddingHead
|